diff options
| author | Alex Schofield <git@ajschof.me> | 2025-02-19 03:50:09 +0000 |
|---|---|---|
| committer | Alex Schofield <git@ajschof.me> | 2025-02-19 03:50:09 +0000 |
| commit | 422acef7a0762089298e9eae9944877e788fd94d (patch) | |
| tree | 3538d69a19e20dad074a1a8600df08d82448a7b4 | |
| parent | 648c3785071adfae163e7ac1321f19188a5e4921 (diff) | |
| download | gdpr-obfuscator-422acef7a0762089298e9eae9944877e788fd94d.tar.gz gdpr-obfuscator-422acef7a0762089298e9eae9944877e788fd94d.zip | |
fix some tests (some are still broken)
| -rw-r--r-- | obfuscator/csv_writer.py | 1 | ||||
| -rw-r--r-- | obfuscator/obfuscate.py | 4 |
2 files changed, 2 insertions, 3 deletions
diff --git a/obfuscator/csv_writer.py b/obfuscator/csv_writer.py index 2bff6e0..de7cd4b 100644 --- a/obfuscator/csv_writer.py +++ b/obfuscator/csv_writer.py @@ -9,7 +9,6 @@ logger = get_logger("CSVWRITER") def create_byte_stream(data: List[Dict[str, str]]) -> bytes: if not data: logger.error("Invalid or empty data was provided to write") - raise output = io.StringIO() diff --git a/obfuscator/obfuscate.py b/obfuscator/obfuscate.py index 73448ce..9cd3a03 100644 --- a/obfuscator/obfuscate.py +++ b/obfuscator/obfuscate.py @@ -13,10 +13,10 @@ def obfuscate( """ if not data: logger.error("Invalid or empty was provided to obfuscate") - raise + raise ValueError("Invalid data provided to obfuscate") if not pii_fields: logger.error("No PII fields provided to obfuscate") - raise + raise ValueError("No PII fields provided to obfuscate") return [ {k: ("***" if k in pii_fields else v) for k, v in record.items()} |
