Skip to content

Commit d81df40

Browse files
committed
add test to make sure dictionaries with missing keys work
1 parent d600dfe commit d81df40

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

pandas/tests/indexes/test_base.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -841,7 +841,6 @@ def test_map_with_series_all_indices(self):
841841
self.assert_numpy_array_equal(expected.values.get_values(), output.values.get_values())
842842
self.assert_equal(expected.name, output.name)
843843

844-
845844
for name in list(set(self.indices.keys()) - set(special)):
846845
cur_index = self.indices[name]
847846
expected = Index(np.arange(len(cur_index), 0, -1))
@@ -861,12 +860,16 @@ def test_map_with_categorical_series(self):
861860
exp = Index(["odd", "even", "odd", np.nan])
862861
self.assert_index_equal(a.map(c), exp)
863862

864-
def test_map_with_series_missing_values(self):
863+
def test_map_with_non_function_missing_values(self):
865864
# GH 12756
866865
expected = Index([2., np.nan, 'foo'])
866+
input = Index([2, 1, 0])
867+
867868
mapper = Series(['foo', 2., 'baz'], index=[0, 2, -1])
868-
output = Index([2, 1, 0]).map(mapper)
869-
self.assert_index_equal(output, expected)
869+
self.assert_index_equal(expected, input.map(mapper))
870+
871+
mapper = {0: 'foo', 2: 2.0, -1: 'baz'}
872+
self.assert_index_equal(expected, input.map(mapper))
870873

871874
def test_append_multiple(self):
872875
index = Index(['a', 'b', 'c', 'd', 'e', 'f'])

0 commit comments

Comments
 (0)