def test_map_by_substring_with_map_values(self): np.testing.assert_equal( map_by_substring(self.arr, ["abc", "a", "bc", ""], case_sensitive=True, match_beginning=False, map_values=None), [0, 1, 2, 0, 3]) np.testing.assert_equal( map_by_substring(self.arr, ["abc", "a", "bc", ""], case_sensitive=True, match_beginning=False, map_values=[0, 1, 2, 3]), [0, 1, 2, 0, 3]) np.testing.assert_equal( map_by_substring(self.arr, ["abc", "a", "bc", ""], case_sensitive=True, match_beginning=False, map_values=[1, 0, 3, 2]), [1, 0, 3, 1, 2]) np.testing.assert_equal( map_by_substring(self.arr, ["abc", "a", "bc", ""], case_sensitive=True, match_beginning=False, map_values=[1, 1, 0, 0]), [1, 1, 0, 1, 0])
def test_map_by_substring(self): np.testing.assert_equal( map_by_substring(self.arr, ["abc", "a", "bc", ""], case_sensitive=True, match_beginning=False), [0, 1, 2, 0, 3]) np.testing.assert_equal( map_by_substring(self.arr, ["abc", "a", "Bc", ""], case_sensitive=True, match_beginning=False), [0, 1, 3, 0, 3]) np.testing.assert_equal( map_by_substring(self.arr, ["abc", "a", "Bc", ""], case_sensitive=False, match_beginning=False), [0, 1, 2, 0, 3]) np.testing.assert_equal( map_by_substring(self.arr, ["abc", "a", "bc", ""], case_sensitive=False, match_beginning=True), [0, 1, 2, 3, 3]) np.testing.assert_equal( map_by_substring(self.arr, ["", ""], False, False), 0) self.assertTrue(np.all(np.isnan( map_by_substring(self.arr, [], False, False))))