Python record_blocks Examples

Example #1

File: tests.py Project: blue-systems-group/project.maybe.rewriter

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

Example #2

File: tests.py Project: blue-systems-group/project.maybe.rewriter

 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)