def test_add_instruction_cjmp(self): block = InstructionBlock() expected_instructions = [] expected_compiled_instructions = [] targets = [(InstructionBlock(), 0), (InstructionBlock(), 1), (InstructionBlock(), 50)] triggers = [Trigger(), Trigger()] LOOKUP = [(0, 0), (1, 0), (1, 1), (0, 1), (2, 0), (1, 0), (0, 1), (0, 1), (0, 0), (1, 0), (2, 1), (2, 1)] for i in LOOKUP: block.add_instruction_cjmp(triggers[i[1]], targets[i[0]]) expected_instructions.append(CJMPInstruction(triggers[i[1]], targets[i[0]], 0)) expected_compiled_instructions = expected_instructions.copy() expected_compiled_instructions.append(STOPInstruction()) self.__verify_block(block, expected_instructions, expected_compiled_instructions)
def test_nested_block_construction(self): main_block = InstructionBlock() expected_instructions = [[], [], [], []] expected_compiled_instructions = [[], [], [], []] blocks = [] waveforms = [DummyWaveform(), DummyWaveform(), DummyWaveform()] main_block.add_instruction_exec(waveforms[0]) expected_instructions[0].append(EXECInstruction(waveforms[0])) block = main_block.create_embedded_block() trigger = Trigger() main_block.add_instruction_cjmp(trigger, block) expected_instructions[0].append(CJMPInstruction(trigger, block, 0)) block.return_ip = InstructionPointer(main_block, len(main_block)) blocks.append(block) block = main_block.create_embedded_block() trigger = Trigger() main_block.add_instruction_cjmp(trigger, block) expected_instructions[0].append(CJMPInstruction(trigger, block, 0)) block.return_ip = InstructionPointer(main_block, len(main_block)) blocks.append(block) WAVEFORM_LOOKUP = [[2, 2, 1, 1],[0, 1, 1, 0, 2, 1]] for i in [0, 1]: block = blocks[i] lookup = WAVEFORM_LOOKUP[i] for id in lookup: waveform = waveforms[id] expected_instructions[i + 1].append(EXECInstruction(waveform)) block.add_instruction_exec(waveform) block = blocks[0].create_embedded_block() blocks[0].add_instruction_cjmp(trigger, block) expected_instructions[1].append(CJMPInstruction(trigger, block, 0)) block.return_ip = InstructionPointer(blocks[0], len(blocks[0])) blocks.append(block) for id in [1, 2, 0, 2]: waveform = waveforms[id] expected_instructions[3].append(EXECInstruction(waveform)) block.add_instruction_exec(waveform) for i in [0, 1, 2, 3]: expected_compiled_instructions[i] = expected_instructions[i].copy() expected_compiled_instructions[0].append(STOPInstruction()) for i in [0, 1, 2]: expected_compiled_instructions[i + 1].append(GOTOInstruction(blocks[i].return_ip.block, blocks[i].return_ip.offset)) positions = [0, None, None, None] positions[3] = len(expected_compiled_instructions[1]) expected_compiled_instructions[1].extend(expected_compiled_instructions[3]) for i in [1, 2]: positions[i] = len(expected_compiled_instructions[0]) expected_compiled_instructions[0].extend(expected_compiled_instructions[i]) positions[3] += positions[1] self.__verify_block(blocks[2], expected_instructions[3], expected_compiled_instructions[3]) self.__verify_block(blocks[1], expected_instructions[2], expected_compiled_instructions[2]) self.__verify_block(blocks[0], expected_instructions[1], expected_compiled_instructions[1]) self.__verify_block(main_block, expected_instructions[0], expected_compiled_instructions[0]) self.assertEqual(positions[3], blocks[2].get_start_address()) self.assertEqual(positions[2], blocks[1].get_start_address()) self.assertEqual(positions[1], blocks[0].get_start_address()) self.assertEqual(positions[0], main_block.get_start_address()) for instruction in main_block.instructions: if isinstance(instruction, GOTOInstruction) or isinstance(instruction, CJMPInstruction): self.assertIsInstance(instruction.target.get_absolute_address(), int)