def test_kind_parameters(self):
dataframe = pd.DataFrame({"value": [1, 2, 3, 4, 5, 6, 7, 8, 9],
"id": [1, 1, 1, 1, 2, 2, 2, 2, 2],
"kind": [0, 1, 0, 1, 0, 1, 0, 1, 0]})
extended_dataframe = dataframe_functions.add_sub_time_series_index(dataframe, 2,
column_id="id",
column_kind="kind")
self.assertEqual(list(extended_dataframe["id"]),
["0,1", "0,1", "0,1", "0,1", "0,2", "0,2", "0,2", "0,2", "1,2"])
assert_series_equal(dataframe["value"], extended_dataframe["value"])
assert_series_equal(dataframe["kind"], extended_dataframe["kind"])
def test_dict_input(self):
dataframe = pd.DataFrame({"value": [1, 2, 3, 4, 5, 6, 7, 8, 9],
"id": [1, 1, 1, 1, 2, 2, 2, 2, 2]})
extended_dataframe = dataframe_functions.add_sub_time_series_index({"1": dataframe}, 2,
column_id="id")
self.assertIn("1", extended_dataframe)
extended_dataframe = extended_dataframe["1"]
self.assertEqual(list(extended_dataframe["id"]),
["0,1", "0,1", "1,1", "1,1", "0,2", "0,2", "1,2", "1,2", "2,2"])
assert_series_equal(dataframe["value"], extended_dataframe["value"])