def test_empty_map(self):
# when
mapped_dict = map_values(times_2, {})
# then
self.assertEqual({}, mapped_dict)
def test_non_empty_map(self):
# when
mapped_dict = map_values(times_2, {'a': 2, 'b': 3})
# then
self.assertEqual({'a': 4, 'b': 6}, mapped_dict)
def _simple_dict_to_dataframe(d):
return pd.DataFrame.from_dict(map_values(lambda x: [x], d))