def __init__(self, transform=None, reverse_transform=None, is_valid=None):
if transform is not None:
self.transform = import_object(transform)
if reverse_transform is not None:
self.reverse_transform = import_object(reverse_transform)
if is_valid is not None:
self.is_valid = import_object(is_valid)
def __init__(self, column, formula, handling_strategy='transform', drop_column=True):
self._column = column
self.constraint_columns = (column,)
self._formula = import_object(formula)
self._drop_column = drop_column
self.rebuild_columns = (column,)
super().__init__(handling_strategy, fit_columns_model=False)
def __init__(self, columns=None, transform=None, reverse_transform=None, is_valid=None):
if isinstance(columns, str):
self._columns = [columns]
else:
self._columns = columns
self.fit_columns_model = False
if transform is not None:
self._transform = import_object(transform)
self.transform = self._run_transform
if reverse_transform is not None:
self._reverse_transform = import_object(reverse_transform)
self.reverse_transform = self._run_reverse_transform
if is_valid is not None:
self._is_valid = import_object(is_valid)
self.is_valid = self._run_is_valid
def test_import_object_class():
"""Test the ``import_object`` function, when importing a class.
The ``import_object`` function is expected to:
- Import a class from its qualifed name.
Input:
- Qualified name of the class.
Output:
- The imported class.
"""
# Run
obj = import_object('sdv.constraints.base.Constraint')
# Assert
assert obj is Constraint
def test_import_object_function():
"""Test the ``import_object`` function, when importing a function.
The ``import_object`` function is expected to:
- Import a function from its qualifed name.
Input:
- Qualified name of the function.
Output:
- The imported function.
"""
# Run
imported = import_object('sdv.constraints.base.import_object')
# Assert
assert imported is import_object
def __init__(self, column, formula, handling_strategy='transform'):
self._column = column
self._formula = import_object(formula)
self._handling_strategy = handling_strategy
super().__init__()