renaming

Author: R-Palazzo

sdv/cag/_utils.py

@@ -210,7 +210,7 @@ def _validate_constraints_single_table(constraints, synthesizer_fitted):
     return constraints
 
 
-def _format_error_message_inequality_constraints(invalid_data, constraint, table_name):
+def _format_error_message_constraint(invalid_data, constraint, table_name):
     """Format the error message for inequality constraints."""
     invalid_rows_str = _format_invalid_values_string(invalid_data, 5)
     raise ConstraintNotMetError(

sdv/cag/fixed_increments.py

@@ -5,7 +5,7 @@
 from sdv._utils import _create_unique_name
 from sdv.cag._errors import ConstraintNotMetError
 from sdv.cag._utils import (
-    _format_error_message_inequality_constraints,
+    _format_error_message_constraint,
     _get_is_valid_dict,
     _remove_columns_from_metadata,
     _validate_table_and_column_names,
@@ -114,7 +114,7 @@ def _validate_constraint_with_data(self, data, metadata):
         if not valid.all():
             table_name = self._get_single_table_name(metadata)
             invalid_rows = data[table_name].loc[~valid, [self.column_name]]
-            _format_error_message_inequality_constraints(invalid_rows, self, table_name)
+            _format_error_message_constraint(invalid_rows, self, table_name)
 
     def _get_updated_metadata(self, metadata):
         """Get the updated metadata after applying the constraint to the metadata.

sdv/cag/inequality.py

@@ -7,7 +7,7 @@
 from sdv._utils import _convert_to_timedelta, _create_unique_name
 from sdv.cag._errors import ConstraintNotMetError
 from sdv.cag._utils import (
-    _format_error_message_inequality_constraints,
+    _format_error_message_constraint,
     _get_is_valid_dict,
     _is_list_of_type,
     _remove_columns_from_metadata,
@@ -146,7 +146,7 @@ def _validate_constraint_with_data(self, data, metadata):
         valid = pd.isna(low) | pd.isna(high) | self._operator(high, low)
         if not valid.all():
             invalid_rows = data.loc[~valid, [self._low_column_name, self._high_column_name]]
-            _format_error_message_inequality_constraints(invalid_rows, self, table_name)
+            _format_error_message_constraint(invalid_rows, self, table_name)
 
     def _get_diff_and_nan_column_names(self, metadata, column_name, table_name):
         """Get the column names for the difference and NaN columns.

sdv/cag/one_hot_encoding.py

@@ -3,7 +3,7 @@
 import numpy as np
 
 from sdv.cag._utils import (
-    _format_error_message_inequality_constraints,
+    _format_error_message_constraint,
     _get_is_valid_dict,
     _is_list_of_type,
     _validate_table_and_column_names,
@@ -74,7 +74,7 @@ def _validate_constraint_with_data(self, data, metadata):
         valid = self._get_valid_table_data(data[table_name])
         if not valid.all():
             invalid_rows = data[table_name].loc[~valid, self._column_names]
-            _format_error_message_inequality_constraints(invalid_rows, self, table_name)
+            _format_error_message_constraint(invalid_rows, self, table_name)
 
     def _fit(self, data, metadata):
         """Fit the constraint.

sdv/cag/range.py

@@ -9,7 +9,7 @@
 from sdv._utils import _convert_to_timedelta, _create_unique_name
 from sdv.cag._errors import ConstraintNotMetError
 from sdv.cag._utils import (
-    _format_error_message_inequality_constraints,
+    _format_error_message_constraint,
     _get_is_valid_dict,
     _is_list_of_type,
     _remove_columns_from_metadata,
@@ -172,7 +172,7 @@ def _validate_constraint_with_data(self, data, metadata):
             invalid_rows = data[table_name].loc[
                 ~valid, [self._low_column_name, self._middle_column_name, self._high_column_name]
             ]
-            _format_error_message_inequality_constraints(invalid_rows, self, table_name)
+            _format_error_message_constraint(invalid_rows, self, table_name)
 
     def _get_diff_and_nan_column_names(self, metadata, table_name):
         """Create unique names for the low, high, and nan component columns."""

tests/unit/cag/test__utils.py

@@ -10,7 +10,7 @@
 from sdv.cag._utils import (
     _convert_to_snake_case,
     _filter_old_style_constraints,
-    _format_error_message_inequality_constraints,
+    _format_error_message_constraint,
     _is_list_of_type,
     _remove_columns_from_metadata,
     _validate_constraints,
@@ -328,8 +328,8 @@ def test__validate_constraints_single_table(mock_validate_constraints):
 
 
 @patch('sdv.cag._utils._format_invalid_values_string')
-def test__format_error_message_inequality_constraints(mock_format_invalid_values_string):
-    """Test `_format_error_message_inequality_constraints` method."""
+def test__format_error_message_constraint(mock_format_invalid_values_string):
+    """Test `_format_error_message_constraint` method."""
     # Setup
     invalid_data = {'row_1': 'value_1', 'row_2': 'value_2'}
     constraint = Mock()
@@ -346,6 +346,6 @@ def test__format_error_message_inequality_constraints(mock_format_invalid_values
 
     # Run and Assert
     with pytest.raises(ConstraintNotMetError, match=expected_error_message):
-        _format_error_message_inequality_constraints(invalid_data, constraint, table_name)
+        _format_error_message_constraint(invalid_data, constraint, table_name)
 
     mock_format_invalid_values_string.assert_called_once_with(invalid_data, 5)