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]))