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__()