1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
from obfuscator.obfuscate import obfuscate
def test_obfuscate_data_with_valid_pii_fields():
data = [
{"student_id": "1234", "name": "John Smith", "course": "Software", "email_address": "j.smith@email.com"},
{"student_id": "5678", "name": "Jane Doe", "course": "Data Science", "email_address": "j.doe@email.com"}
]
pii_fields = ["name", "email_address"]
expected = [
{"student_id": "1234", "name": "***", "course": "Software", "email_address": "***"},
{"student_id": "5678", "name": "***", "course": "Data Science", "email_address": "***"}
]
result = obfuscate(data, pii_fields)
assert result == expected
def test_obfuscate_data_with_missing_pii_field():
data = [
{"student_id": "1234", "name": "John Smith", "course": "Software"},
{"student_id": "5678", "name": "Jane Doe", "course": "Data Science", "email_address": "j.doe@email.com"}
]
pii_fields = ["name", "email_address"]
expected = [
{"student_id": "1234", "name": "***", "course": "Software"},
{"student_id": "5678", "name": "***", "course": "Data Science", "email_address": "***"}
]
result = obfuscate(data, pii_fields)
assert result == expected
def test_obfuscate_data_with_no_data():
data = []
pii_fields = ["name", "email_address"]
expected = []
result = obfuscate(data, pii_fields)
assert result == expected
def test_obfuscate_data_with_empty_pii_fields():
data = [
{"student_id": "1234", "name": "John Smith", "course": "Software", "email_address": "j.smith@email.com"}
]
pii_fields = []
expected = data.copy()
result = obfuscate(data, pii_fields)
assert result == expected
|
