From 6bb31b2544bd091bf1c9c2007db725ebefa11298 Mon Sep 17 00:00:00 2001 From: Scott Sanderson Date: Wed, 7 Jun 2017 14:16:17 -0400 Subject: [PATCH] TEST: Test map ignores missing with None. --- tests/test_labelarray.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/test_labelarray.py b/tests/test_labelarray.py index 6d5681b6..de73eed9 100644 --- a/tests/test_labelarray.py +++ b/tests/test_labelarray.py @@ -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(