Change enum type to string in test

CDCgov · Feb 7, 2025 · 0582e82 · 0582e82
1 parent 769d2ec
commit 0582e82
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/unit/schemas/test_pii.py b/tests/unit/schemas/test_pii.py
@@ -266,11 +266,11 @@ def test_feature_iter(self):
         record = pii.PIIRecord(race="asked but unknown")
         assert list(record.feature_iter(pii.Feature(attribute=pii.FeatureAttribute.RACE))) == []
         record = pii.PIIRecord(race="asian")
-        assert list(record.feature_iter(pii.Feature(attribute=pii.FeatureAttribute.RACE))) == [pii.Race.ASIAN]
+        assert list(record.feature_iter(pii.Feature(attribute=pii.FeatureAttribute.RACE))) == ["ASIAN"]
         record = pii.PIIRecord(race="african american")
-        assert list(record.feature_iter(pii.Feature(attribute=pii.FeatureAttribute.RACE))) == [pii.Race.BLACK]
+        assert list(record.feature_iter(pii.Feature(attribute=pii.FeatureAttribute.RACE))) == ["BLACK"]
         record = pii.PIIRecord(race="white")
-        assert list(record.feature_iter(pii.Feature(attribute=pii.FeatureAttribute.RACE))) == [pii.Race.WHITE]
+        assert list(record.feature_iter(pii.Feature(attribute=pii.FeatureAttribute.RACE))) == ["WHITE"]
 
 
     def test_blocking_keys_invalid(self):