def test_create_label_tensors_fails_with_unknown_label(self): tensorizer = LabelTensorizer(column="label") init = tensorizer.initialize() init.send(None) # kick for row in self.data.train: init.send(row) init.close() batch = [ {"label": types.Label("non/existent")}, {"label": types.Label("alarm/set_alarm")}, ] with self.assertRaises(Exception): tensorizer.create_training_tensors(batch)
def test_create_label_tensors(self): tensorizer = LabelTensorizer(column="label") init = tensorizer.initialize() init.send(None) # kick for row in self.data.train: init.send(row) init.close() batch = [ {"label": types.Label("weather/find")}, {"label": types.Label("alarm/set_alarm")}, ] tensor = tensorizer.create_training_tensors(batch) self.assertEqual((2,), tensor.size()) self.assertEqual([6, 1], tensor.tolist())