Skip to content

Commit fad1570

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

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
@@ -840,7 +840,6 @@ def test_map_with_series_all_indices(self):
840840
self.assert_numpy_array_equal(expected.values.get_values(), output.values.get_values())
841841
self.assert_equal(expected.name, output.name)
842842

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

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

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

0 commit comments

Comments
 (0)