fix tests by resolving imports and removing redundent parameters

3 files changed, 14 insertions, 14 deletions

diff --git a/test/test_obfuscator.py b/test/test_obfuscator.py
index c77b6b4..66d0312 100644
--- a/test/test_obfuscator.py
+++ b/test/test_obfuscator.py
@@ -1,4 +1,4 @@
-from obfuscator.obfuscate import obfuscate
+from gdpr_obfuscator.obfuscate import obfuscate_data
 
 
 def test_obfuscate_data_with_valid_pii_fields():
@@ -32,7 +32,7 @@ def test_obfuscate_data_with_valid_pii_fields():
         },
     ]
 
-    result = obfuscate(data, pii_fields)
+    result = obfuscate_data(data, pii_fields)
     assert result == expected
 
 
@@ -57,7 +57,7 @@ def test_obfuscate_data_with_missing_pii_field():
         },
     ]
 
-    result = obfuscate(data, pii_fields)
+    result = obfuscate_data(data, pii_fields)
     assert result == expected
 
 
@@ -66,7 +66,7 @@ def test_obfuscate_data_with_no_data():
     pii_fields = ["name", "email_address"]
     expected = []
 
-    result = obfuscate(data, pii_fields)
+    result = obfuscate_data(data, pii_fields)
     assert result == expected
 
 
@@ -82,5 +82,5 @@ def test_obfuscate_data_with_empty_pii_fields():
     pii_fields = []
     expected = data.copy()
 
-    result = obfuscate(data, pii_fields)
+    result = obfuscate_data(data, pii_fields)
     assert result == expected
diff --git a/test/test_read.py b/test/test_read.py
index de425ce..81ba053 100644
--- a/test/test_read.py
+++ b/test/test_read.py
@@ -1,9 +1,9 @@
 import boto3
 from moto import mock_aws
-from obfuscator.read import DataReader
+from gdpr_obfuscator.read import DataReader
 import pytest
 
-reader = DataReader(log_level="DEBUG")
+reader = DataReader()
 
 
 def test_empty_csv_should_return_no_content():
diff --git a/test/test_write.py b/test/test_write.py
index 4929b06..9a66088 100644
--- a/test/test_write.py
+++ b/test/test_write.py
@@ -1,8 +1,8 @@
 import io
 import csv
-from obfuscator.write import DataWriter
+from gdpr_obfuscator import Utilities
 
-writer = DataWriter()
+utils = Utilities()
 
 
 def csv_bytes_to_list(csv_bytes: bytes):
@@ -17,14 +17,14 @@ def test_create_byte_stream_valid_data():
         {"student_id": "1234", "name": "Student 1", "course": "Course 1"},
         {"student_id": "5678", "name": "Student 2", "course": "Course 2"},
     ]
-    csv_bytes = writer.create_byte_stream(data)
+    csv_bytes = utils.create_byte_stream(data)
     result = csv_bytes_to_list(csv_bytes)
     assert result == data
 
 
 def test_create_byte_stream_empty_data():
     data = []
-    csv_bytes = writer.create_byte_stream(data)
+    csv_bytes = utils.create_byte_stream(data)
     assert csv_bytes == b""
 
 
@@ -33,7 +33,7 @@ def test_create_byte_stream_handles_quoted_fields():
         {"student_id": "1234", "name": 'Student "One"', "course": "Course, A"},
         {"student_id": "5678", "name": 'Student "Two"', "course": "Course, B"},
     ]
-    csv_bytes = writer.create_byte_stream(data)
+    csv_bytes = utils.create_byte_stream(data)
     result = csv_bytes_to_list(csv_bytes)
     assert result == data
 
@@ -43,7 +43,7 @@ def test_create_byte_stream_consistent_header_order():
         {"student_id": "1234", "name": "Alice", "course": "Math"},
         {"student_id": "5678", "name": "Bob", "course": "Science"},
     ]
-    csv_bytes = writer.create_byte_stream(data)
+    csv_bytes = utils.create_byte_stream(data)
     csv_string = csv_bytes.decode("utf-8")
     header_line = csv_string.splitlines()[0]
     expected_header = ",".join(data[0].keys())
@@ -55,6 +55,6 @@ def test_create_byte_stream_special_characters():
         {"student_id": "1234", "name": "Student 1", "course": "Line1\nLine2"},
         {"student_id": "5678", "name": "Student 2", "course": "Value with, comma"},
     ]
-    csv_bytes = writer.create_byte_stream(data)
+    csv_bytes = utils.create_byte_stream(data)
     result = csv_bytes_to_list(csv_bytes)
     assert result == data