def test_record_blocks(self): statements = rewrite.record_blocks(self.test_input) self.assertEqual(len(statements), 6) self.assertEqual(len([s for s in statements.values() if s.is_block]), 6) for label, statement in statements.items(): self.assertEqual(statement.label, self.answers[label]['label']) self.assertEqual(len(statement.alternatives), self.answers[label]['alternative_count'])
def test_dump_statements(self): statements = rewrite.record_assignments(self.test_input) statements = rewrite.record_blocks(self.test_input, statements) json_statements = json.loads(rewrite.dump_statements(self.test_input, statements)) self.assertEqual(json_statements['package'], "testing_inputs.maybe") self.assertEqual(len(json_statements['statements']), 8)