TST: add tests for dict with bool keys

Author: cpcloud

pandas/tests/test_frame.py

@@ -8003,6 +8003,12 @@ def test_replace_bool_with_bool(self):
         expected = DataFrame(np.ones((2, 2), dtype=bool))
         tm.assert_frame_equal(result, expected)
 
+    def test_replace_with_dict_with_bool_keys(self):
+        df = DataFrame({0: [True, False], 1: [False, True]})
+        result = df.replace({'asdf': 'asdb', True: 'yes'})
+        expected = DataFrame({0: ['yes', False], 1: [False, 'yes']})
+        tm.assert_frame_equal(expected, result)
+
     def test_combine_multiple_frames_dtypes(self):
         from pandas import concat
 

pandas/tests/test_series.py

@@ -5291,7 +5291,6 @@ def test_replace(self):
         result = ser.replace(Timestamp('20130103'), Timestamp('20120101'))
         assert_series_equal(result, expected)
 
-
     def test_replace_with_single_list(self):
         ser = Series([0, 1, 2, 3, 4])
         result = ser.replace([1,2,3])
@@ -5307,6 +5306,7 @@ def test_replace_with_single_list(self):
             s.replace([1,2,3],inplace=True,method='crash_cymbal')
         assert_series_equal(s, ser)
 
+
     def test_replace_mixed_types(self):
         s = Series(np.arange(5),dtype='int64')
 
@@ -5367,6 +5367,12 @@ def test_replace_bool_with_bool(self):
         expected = Series([False] * len(s))
         tm.assert_series_equal(expected, result)
 
+    def test_replace_with_dict_with_bool_keys(self):
+        s = Series([True, False, True])
+        result = s.replace({'asdf': 'asdb', True: 'yes'})
+        expected = Series(['yes', False, 'yes'])
+        tm.assert_series_equal(expected, result)
+
     def test_asfreq(self):
         ts = Series([0., 1., 2.], index=[datetime(2009, 10, 30),
                                          datetime(2009, 11, 30),