Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
frances-h committed Aug 14, 2023
1 parent b4a2e22 commit 6adb261
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions tests/unit/data_processing/test_data_processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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'
Expand Down

0 comments on commit 6adb261

Please sign in to comment.