def testGetStages(self): pipeline = Pipeline("L.D F1, 0(R2)\nADD.D F2, F3, F4\nMUL.D F2, F3, F4\n", False) self.assertEqual(pipeline.get_stages(pipeline.Code[1][0]), Pipeline.data_stages) self.assertEqual(pipeline.get_stages(pipeline.Code[2][0]), Pipeline.add_stages) self.assertEqual(pipeline.get_stages(pipeline.Code[3][0]), Pipeline.mult_stages)