fix test

sdv-dev · Jun 28, 2023 · bad016d · bad016d
1 parent 8501d25
commit bad016d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/unit/transformers/test_datetime.py b/tests/unit/transformers/test_datetime.py
@@ -444,6 +444,7 @@ def test__reverse_transform_datetime_format_with_nans(self):
 
  # Assert
  expected = pd.Series(['Jan 1, 2020', 'Feb 1, 2020', np.nan])
+ pd.testing.assert_series_equal(output, expected)
 
  def test__reverse_transform_only_nans(self):
  """Test the ``_reverse_transform`` method returns the correct datetime format with nans."""
@@ -458,6 +459,7 @@ def test__reverse_transform_only_nans(self):
 
  # Assert
  expected = pd.Series([np.nan, np.nan, np.nan])
+ pd.testing.assert_series_equal(output, expected)
 
 
 class TestOptimizedTimestampEncoder: