def test_append_max_age(self):
df = pd.DataFrame(index=t_date_range, data={"rx": rx_list, "tx": tx_list})
cml_ch_shortened = ComlinkChannel(data=df.iloc[:100, :], frequency=f)
cml_ch_rest_of_data = ComlinkChannel(data=df.iloc[100:, :], frequency=f)
cml_ch_shortened.append_data(cml_ch_rest_of_data, max_age="60min")
assert len(cml_ch_shortened.data) == len(
df.iloc[df.index > (df.index[-1] - pd.Timedelta("60min")), :]
)
assert cml_ch_shortened.data.index[-1] == df.index[-1]
def test_append_max_length(self):
df = pd.DataFrame(index=t_date_range, data={"rx": rx_list, "tx": tx_list})
cml_ch_shortened = ComlinkChannel(data=df.iloc[:100, :], frequency=f)
cml_ch_rest_of_data = ComlinkChannel(data=df.iloc[100:, :], frequency=f)
cml_ch_shortened.append_data(cml_ch_rest_of_data, max_length=10)
assert len(cml_ch_shortened.data) == 10
assert cml_ch_shortened.data.index[-1] == df.index[-1]
assert cml_ch_shortened.data.index[0] == df.index[-10]
assert cml_ch_shortened.data.tx[-1] == df.tx[-1]
assert cml_ch_shortened.data.tx[0] == df.tx[-10]
def test_append_no_kwargs(self):
df = pd.DataFrame(index=t_date_range, data={"rx": rx_list, "tx": tx_list})
cml_ch_full = ComlinkChannel(data=df, frequency=f)
cml_ch_shortened = ComlinkChannel(data=df.iloc[:100, :], frequency=f)
cml_ch_rest_of_data = ComlinkChannel(data=df.iloc[100:, :], frequency=f)
assert len(cml_ch_full.data) == len(t_date_range)
assert len(cml_ch_shortened.data) == len(t_date_range[:100])
assert len(cml_ch_rest_of_data.data) == len(t_date_range[100:])
cml_ch_shortened.append_data(cml_ch_rest_of_data)
assert_comlink_channel_equal(cml_ch_shortened, cml_ch_full)
