def test_session_data_for_ids(model_data: RasaModelData):
filtered_data = model_data._data_for_ids(model_data.data, np.array([0, 1]))
for values in filtered_data.values():
for data in values.values():
for v in data:
assert v.shape[0] == 2
key = model_data.keys()[0]
sub_key = model_data.keys(key)[0]
assert np.all(
np.array(filtered_data[key][sub_key][0][0]) == np.array(
model_data.get(key, sub_key)[0][0]))
assert np.all(
np.array(filtered_data[key][sub_key][0][1]) == np.array(
model_data.get(key, sub_key)[0][1]))
def test_session_data_for_ids(model_data: RasaModelData):
filtered_data = model_data._data_for_ids(model_data.data, np.array([0, 1]))
for values in filtered_data.values():
for v in values:
assert v.shape[0] == 2
k = list(model_data.keys())[0]
assert np.all(np.array(filtered_data[k][0][0]) == np.array(model_data.get(k)[0][0]))
assert np.all(np.array(filtered_data[k][0][1]) == np.array(model_data.get(k)[0][1]))
