test_team_tech_baddies.py

"""Unit test file for team tech_baddies"""
import unittest
from pii_scan import analyze_text, show_aggie_pride  # noqa 


class TestTeam_tech_baddies(unittest.TestCase):
    """Test team tech_baddies PII functions"""
    def test_show_aggie_pride(self):
        """Test to make sure Aggie Pride is shown correctly"""
        self.assertEqual(show_aggie_pride(), "Aggie Pride - Worldwide")

    def test_in_aadhaar(self):
        """Test IN_AADHAAR functionality"""
        prefix = ['3123']
        mid = ['4567']
        suffix = ['8909']

        # Positive test cases
        for p in prefix:
            for m in mid:
                for s in suffix:
                    aadhaar = ''.join([p, m, s])
                    
                    # Check with context score 
                    result = analyze_text('My Aadhaar is ' + aadhaar, ['IN_AADHAAR'])
                    print("Result with context:", result)
                    
                    if result:
                        self.assertEqual('IN_AADHAAR', result[0].entity_type)
                        self.assertEqual(1.0, result[0].score)
                    else:
                        print("No Aadhaar detected with context for:", aadhaar)

                    # Check without context
                    result = analyze_text('My abc is ' + aadhaar, ['IN_AADHAAR'])
                    print("Result without context:", result)

                    if result:
                        self.assertEqual('IN_AADHAAR', result[0].entity_type)
                        self.assertEqual(1.0, result[0].score)
                    else:
                        print("No Aadhaar detected without context for:", aadhaar)

        # Negative test case (invalid Aadhaar number)
        result = analyze_text('My Aadhaar is 000000000000', ['IN_AADHAAR'])
        print("Negative test result:", result)
        self.assertListEqual([], result)

    def test_in_pan(self):
        """Test IN_PAN functionality"""
        # Positive test cases
        # check context
        result = analyze_text('My IN_PAN is AAAPZ1234C', ['IN_PAN'])
        print(result)
        self.assertEqual('IN_PAN', result[0].entity_type)
        self.assertEqual(1.0, result[0].score)

        # check no context
        result = analyze_text('My IN_PAN is 12APZ1234C', ['IN_PAN'])
        print(result)
        self.assertEqual('IN_PAN', result[0].entity_type)
        self.assertEqual(0.4, result[0].score)

        # Negative test cases
        result = analyze_text('My IN_PAN is 0000000000', ['IN_PAN'])

        self.assertListEqual([], result)

    def test_in_passport(self):
        """Test IN_PASSPORT functionality"""
        # postive test cases
        alpha, numeric = ['A','Z'], ['12345678','87654321']
        for letter in alpha:
            for num in numeric:
                passport = ''.join([letter,num])

                # check with context
                result = analyze_text('My passport number is ' + passport,['IN_PASSPORT'])
                print(result)
                self.assertEqual('IN_PASSPORT', result[0].entity_type)
                self.assertAlmostEqual(0.45, result[0].score, places=2)

                # check with no context
                result = analyze_text('My info is ' + passport,['IN_PASSPORT'])
                self.assertEqual('IN_PASSPORT', result[0].entity_type)
                self.assertEqual(0.1, result[0].score)

        # negative test case - no result found
        result = analyze_text('My passport number is A00000000',['IN_PASSPORT'])
        self.assertListEqual([], result)

    def test_in_vehicle_registration(self):
        """Test IN_VEHICLE_REGISTRATION functionality"""
        # Positive test case with context
        registration = "KA01AB1234"
        result = analyze_text('My registration number is ' + registration, ['IN_VEHICLE_REGISTRATION'])
        print(result)
        self.assertEqual('IN_VEHICLE_REGISTRATION', result[0].entity_type)
        self.assertAlmostEqual(1.0, result[0].score, places=2)

        # Positive test case without context
        result = analyze_text('My info is ' + registration, ['IN_VEHICLE_REGISTRATION'])
        print(result)
        self.assertEqual('IN_VEHICLE_REGISTRATION', result[0].entity_type)
        self.assertAlmostEqual(1.0, result[0].score, places=2)

        # Negative test case - invalid registration number
        result = analyze_text('My registration number is XX00ZZ0000', ['IN_VEHICLE_REGISTRATION'])
        print(result)
        self.assertListEqual([], result)
    def test_in_voter(self):
        """Test IN_VOTER functionality"""
        begin = ['AAA']
        ending = ['0000000']

        # Positive Test Case
        for b in begin:
            for e in ending:
                number = ''.join([b,e])
                # Check context score should be 0.75
                result = analyze_text('My Voter ID is ' + number, ['IN_VOTER'])
                print(result)
                self.assertEqual('IN_VOTER', result[0].entity_type)
                self.assertEqual(0.75, result[0].score)

                # Check no context
                result = analyze_text('My number is ' + number, ['IN_VOTER'])
                self.assertEqual('IN_VOTER', result[0].entity_type)
                self.assertEqual(0.4, result[0].score)

        # Negative Test Case
        result = analyze_text('My Voter ID is AAA00000', ['IN_VOTER'])
        self.assertListEqual([], result)

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