From 0d05e4f9ca4ca1d2a60802df02f8037886a797fc Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Fri, 21 Feb 2025 03:27:41 +0000 Subject: fix tests by resolving imports and removing redundent parameters --- test/test_write.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'test/test_write.py') 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 -- cgit v1.2.3