diff options
| -rw-r--r-- | gdpr_obfuscator/__init__.py | 4 | ||||
| -rw-r--r-- | gdpr_obfuscator/read.py | 2 | ||||
| -rw-r--r-- | test/test_read.py | 4 |
3 files changed, 5 insertions, 5 deletions
diff --git a/gdpr_obfuscator/__init__.py b/gdpr_obfuscator/__init__.py index 694c512..6c0e47e 100644 --- a/gdpr_obfuscator/__init__.py +++ b/gdpr_obfuscator/__init__.py @@ -1,4 +1,4 @@ -from .read import DataReader +from .read import FileHandler from .obfuscate import obfuscate_data from .utils import Utilities from typing import List @@ -6,7 +6,7 @@ from typing import List class Obfuscator: def __init__(self): - self.reader = DataReader() + self.reader = FileHandler() self.utils = Utilities() def process_s3(self, input: str) -> bytes: diff --git a/gdpr_obfuscator/read.py b/gdpr_obfuscator/read.py index 27f2a3d..a3761ae 100644 --- a/gdpr_obfuscator/read.py +++ b/gdpr_obfuscator/read.py @@ -5,7 +5,7 @@ from typing import List, Dict from .utils import Utilities -class DataReader: +class FileHandler: """ A class to read CSV data from a local file, S3 object, or string. Near the project completion, support for JSON/Parquet files will be added. diff --git a/test/test_read.py b/test/test_read.py index 81ba053..4cd376f 100644 --- a/test/test_read.py +++ b/test/test_read.py @@ -1,9 +1,9 @@ import boto3 from moto import mock_aws -from gdpr_obfuscator.read import DataReader +from gdpr_obfuscator.read import FileHandler import pytest -reader = DataReader() +reader = FileHandler() def test_empty_csv_should_return_no_content(): |