def testIdentityGenerator(self):
identity_problem = algorithmic.AlgorithmicIdentityBinary40()
counter = 0
for d in identity_problem.generator(3, 8, 10):
counter += 1
self.assertEqual(d["inputs"], d["targets"])
self.assertEqual(counter, 10)
def generate_data(self, data_dir, _):
identity_problem = algorithmic.AlgorithmicIdentityBinary40()
generator_utils.generate_files(
identity_problem.generator(self.num_symbols, 40, 100000),
self.training_filepaths(data_dir, 1, shuffled=True), 100)
generator_utils.generate_files(
identity_problem.generator(self.num_symbols, 400, 10000),
self.dev_filepaths(data_dir, 1, shuffled=True), 100)
