Skip to content

Commit 109a3d6

Browse files
committed
Make test reflect what's in master
1 parent a4c916b commit 109a3d6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

category_encoders/tests/test_basen.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,9 +115,9 @@ def test_inverse_transform_BothFieldsAreReturnNanWithNan_ExpectValueError(self):
115115
with warnings.catch_warnings(record=True) as w:
116116
enc.inverse_transform(result)
117117

118-
self.assertEqual(2, len(w))
118+
self.assertEqual(1, len(w))
119119
self.assertEqual('inverse_transform is not supported because transform impute '
120-
'the unknown category nan when encode city', str(w[1].message))
120+
'the unknown category nan when encode city', str(w[0].message))
121121

122122
def test_inverse_transform_HaveMissingAndNoUnknown_ExpectInversed(self):
123123
train = pd.DataFrame({'city': ['chicago', np.nan]})

0 commit comments

Comments
 (0)