From 6adb2616ca9dcaa533fa08884e9a9b935181b965 Mon Sep 17 00:00:00 2001 From: Frances Hartwell Date: Thu, 10 Aug 2023 10:58:45 -0400 Subject: [PATCH] lint --- tests/unit/data_processing/test_data_processor.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/unit/data_processing/test_data_processor.py b/tests/unit/data_processing/test_data_processor.py index 228718670..7587d8160 100644 --- a/tests/unit/data_processing/test_data_processor.py +++ b/tests/unit/data_processing/test_data_processor.py @@ -6,7 +6,8 @@ import pytest from rdt.errors import ConfigNotSetError from rdt.errors import NotFittedError as RDTNotFittedError -from rdt.transformers import AnonymizedFaker, FloatFormatter, IDGenerator, LabelEncoder, UnixTimestampEncoder +from rdt.transformers import ( + AnonymizedFaker, FloatFormatter, IDGenerator, LabelEncoder, UnixTimestampEncoder) from sdv.constraints.errors import MissingConstraintColumnError from sdv.constraints.tabular import Positive, ScalarRange @@ -1142,12 +1143,12 @@ def test__create_config(self): assert isinstance(id_no_regex_transformer, IDGenerator) assert id_no_regex_transformer.prefix == 'sdv-id-' assert id_no_regex_transformer.starting_value == 0 - + id_numeric_transformer = config['transformers']['id_numeric'] assert isinstance(id_numeric_transformer, IDGenerator) - assert id_numeric_transformer.prefix == None + assert id_numeric_transformer.prefix is None assert id_numeric_transformer.starting_value == 0 - + id_column_transformer = config['transformers']['id_column'] assert isinstance(id_column_transformer, AnonymizedFaker) assert id_column_transformer.function_name == 'bothify'