def test_easy_softmax_data_generator_valid_data(self): n_classes, n_examples = 5, 10 X, y = toy_datasets.easy_softmax_data_generator(n_classes, n_examples) actual = True for example, target in zip(X, y): if example[target] != 1: actual = False self.assertTrue(actual)
def test_easy_softmax_data_generator_data_len(self): n_classes, n_examples = 7, 10 X, y = toy_datasets.easy_softmax_data_generator(n_classes, n_examples) self.assertTrue(len(X) == len(y))