We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a4c916b commit 109a3d6Copy full SHA for 109a3d6
category_encoders/tests/test_basen.py
@@ -115,9 +115,9 @@ def test_inverse_transform_BothFieldsAreReturnNanWithNan_ExpectValueError(self):
115
with warnings.catch_warnings(record=True) as w:
116
enc.inverse_transform(result)
117
118
- self.assertEqual(2, len(w))
+ self.assertEqual(1, len(w))
119
self.assertEqual('inverse_transform is not supported because transform impute '
120
- 'the unknown category nan when encode city', str(w[1].message))
+ 'the unknown category nan when encode city', str(w[0].message))
121
122
def test_inverse_transform_HaveMissingAndNoUnknown_ExpectInversed(self):
123
train = pd.DataFrame({'city': ['chicago', np.nan]})
0 commit comments