def test__validate_conditions_with_conditions_invalid_column(self):
"""Test the `BaseTabularModel._validate_conditions` method with an invalid column.
When a condition has an invalid column, expect a ValueError.
Input:
- Conditions DataFrame with an invalid column.
Side Effects:
- A ValueError is thrown.
"""
# Setup
model = Mock(spec_set=CTGAN)
metadata_mock = Mock()
metadata_mock.get_fields.return_value = {'cola': {}}
model._metadata = metadata_mock
conditions = pd.DataFrame([{'colb': 'a'}] * 5)
# Run and Assert
with pytest.raises(
ValueError,
match=(
'Unexpected column name `colb`. '
'Use a column name that was present in the original data.'
)):
BaseTabularModel._validate_conditions(model, conditions)
def test__validate_conditions_with_conditions_valid_columns(self):
"""Test the `BaseTabularModel._validate_conditions` method with valid columns.
Expect no error to be thrown.
Input:
- Conditions DataFrame contains only valid columns.
"""
# Setup
model = Mock(spec_set=CTGAN)
metadata_mock = Mock()
metadata_mock.get_fields.return_value = {'cola': {}}
model._metadata = metadata_mock
conditions = pd.DataFrame([{'cola': 'a'}] * 5)
# Run and Assert
BaseTabularModel._validate_conditions(model, conditions)
