mirror of
https://github.com/wassname/catalyst.git
synced 2026-07-29 11:18:20 +08:00
TEST: Test map returning None.
This commit is contained in:
@@ -163,7 +163,12 @@ class LabelArrayTestCase(ZiplineTestCase):
|
||||
|
||||
# Should work.
|
||||
la = LabelArray(self.strs, missing_value=None)
|
||||
la.map(lambda x: None)
|
||||
result = la.map(lambda x: None)
|
||||
|
||||
check_arrays(
|
||||
result,
|
||||
LabelArray(np.full_like(self.strs, None), missing_value=None),
|
||||
)
|
||||
|
||||
la = LabelArray(self.strs, missing_value="__MISSING__")
|
||||
with self.assertRaises(TypeError):
|
||||
|
||||
Reference in New Issue
Block a user