wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-06-29 13:13:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

TEST: Test map ignores missing with None.

Browse Source
This commit is contained in:
Scott Sanderson
2017-06-07 14:16:17 -04:00
parent e995e6f2ed
commit cfe4df8f2b
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_labelarray.py
+5 -4
View File
@@ -133,15 +133,16 @@ class LabelArrayTestCase(ZiplineTestCase):
assert_equal(numpy_transformed, la_transformed)
def test_map_ignores_missing_value(self):
data = np.array(['A', 'B', 'C'], dtype=object)
la = LabelArray(data, missing_value='A')
@parameter_space(missing=['A', None])
def test_map_ignores_missing_value(self, missing):
data = np.array([missing, 'B', 'C'], dtype=object)
la = LabelArray(data, missing_value=missing)
def increment_char(c):
return chr(ord(c) + 1)
result = la.map(increment_char)
expected = LabelArray(['A', 'C', 'D'], missing_value='A')
expected = LabelArray([missing, 'C', 'D'], missing_value=missing)
assert_equal(result.as_string_array(), expected.as_string_array())
@parameter_space(