test_team_sei.py

import unittest
from pii_team_sei import *


class SEITestCases(unittest.TestCase):
    def test_find_us_phone_number(self):
        results_list = find_us_phone_number('My phone number is 123-456-7890')
        self.assertEqual(results_list[0], '123-456-7890')  # add assertion here

    def test_find_visa_mastercard(self):
        #Not matching credit card number
        results_list = find_visa_mastercard('My credit card number is 1234-5678-9012-3456')
        self.assertEqual(results_list, [])

        #Matching Visa credit card number
        results_list = find_visa_mastercard('My credit card number is 4234-5678-9012-3456')
        self.assertEqual(results_list[0], '4234-5678-9012-3456')

        #Matching Mastercard credit card number
        results_list = find_visa_mastercard('My credit card number is 5234-5678-9012-3456')
        self.assertEqual(results_list[0], '5234-5678-9012-3456')


    def test_find_amex(self):
        results_list = find_amex('My credit card number is 1234-567890-12345')
        self.assertEqual(results_list[0], '1234-567890-12345')

    def test_find_us_ssn(self):
        results_list = find_us_ssn('My social security number is 123-45-6789')
        self.assertEqual(results_list[0], '123-45-6789')

    def test_find_email(self):
        results_list = find_email('My email address is jim.jones@jones.com')
        self.assertEqual(results_list[0], 'jim.jones@jones.com')

    def test_find_instagram_handle(self):
        results_list = find_instagram_handle('My instagram handle is @jimjones')
        self.assertEqual(results_list[0], '@jimjones')


if __name__ == '__main__':
    unittest.main()