Newer
Older
from src.optimize.redundancies import remove_redundant_jumps
from src.optimize import optimize_block
class TestOptimize(unittest.TestCase):
def setUp(self):
self.foo = S('command', 'foo')
self.bar = S('command', 'bar')
def tearDown(self):
del self.foo
del self.bar
def test_optimize_block_movaa(self):
block = B([self.foo,
S('command', 'move', '$regA', '$regA'),
self.assertEquals(block.statements, [self.foo, self.bar])
def test_optimize_block_movab(self):
move = S('command', 'move', '$regA', '$regB')
block = B([self.foo,
self.assertEquals(block.statements, [self.foo, move, self.bar])
def test_optimize_block_movinst_true(self):
block = B([self.foo,
S('command', 'move', '$regA', '$regB'),
S('command', 'addu', '$regA', '$regA', 2),
self.assertEquals(block.statements, [self.foo,
S('command', 'addu', '$regA', '$regB', 2),
def test_optimize_block_movinst_false(self):
statements = [self.foo, \
S('command', 'move', '$regA', '$regB'), \
S('command', 'addu', '$regA', '$regC', 2), \
block = B(statements)
self.assertEquals(block.statements, statements)
def test_optimize_block_instr_mov_jal_true(self):
block = B([self.foo,
S('command', 'addu', '$regA', '$regC', 2),
S('command', 'move', '$4', '$regA'),
S('command', 'jal', 'L1'),
self.assertEquals(block.statements, [self.foo,
S('command', 'addu', '$4', '$regC', 2),
S('command', 'jal', 'L1'),
def test_optimize_block_instr_mov_jal_false(self):
arguments = [self.foo, \
S('command', 'addu', '$regA', '$regC', 2), \
S('command', 'move', '$3', '$regA'), \
S('command', 'jal', 'L1'), \
self.bar]
block = B(arguments)
self.assertEquals(block.statements, arguments)
def test_optimize_block_sw_ld_true(self):
block = B([self.foo,
S('command', 'sw', '$regA', '$regB'),
S('command', 'lw', '$regA', '$regB'),
self.assertEquals(block.statements, [self.foo,
S('command', 'sw', '$regA', '$regB'),
def test_optimize_block_sw_ld_false(self):
arguments = [self.foo, \
S('command', 'sw', '$regA', '$regB'), \
S('command', 'lw', '$regD', '$regC'), \
block = B(arguments)
self.assertEquals(block.statements, arguments)
def test_optimize_block_shift_true(self):
block = B([self.foo,
S('command', 'sll', '$regA', '$regA', 0),
optimize_block(block)
self.assertEquals(block.statements, [self.foo, self.bar])
def test_optimize_block_shift_false(self):
arguments = [self.foo, \
S('command', 'sll', '$regA', '$regB', 0), \
optimize_block(block)
self.assertEquals(block.statements, arguments)
arguments2 = [self.foo, \
S('command', 'sll', '$regA', '$regA', 1), \
optimize_block(block2)
self.assertEquals(block2.statements, arguments2)
def test_optimize_block_add_lw_true(self):
block = B([self.foo,
S('command', 'addu', '$regA', '$regA', 10),
optimize_block(block)
self.assertEquals(block.statements, [self.foo,
S('command', 'lw', '$regB', '10($regA)'),
def test_optimize_block_add_lw_false(self):
arguments = [self.foo, \
S('command', 'addu', '$regA', '$regA', 10), \
S('command', 'lw', '$regB', '0($regC)'), \
optimize_block(block)
arguments2 = [self.foo, \
S('command', 'addu', '$regA', '$regB', 10), \
S('command', 'lw', '$regB', '0($regA)'), \
arguments3 = [self.foo, \
S('command', 'addu', '$regA', '$regA', 10), \
S('command', 'lw', '$regB', '1($regA)'), \
optimize_block(block3)
self.assertEquals(block.statements, arguments)
self.assertEquals(block2.statements, arguments2)
self.assertEquals(block3.statements, arguments3)
def test_remove_redundant_jumps_beq_j_true(self):
block = B([self.foo,
S('command', 'beq', '$regA', '$regB', '$Lx'),
S('command', 'j', '$Ly'),
S('label', '$Lx'),
remove_redundant_jumps(block)
self.assertEquals(block.statements, [self.foo,
S('command', 'bne', '$regA', '$regB', '$Ly'),
S('label', '$Lx'),
def test_remove_redundant_jumps_beq_j_false(self):
arguments = [self.foo, \
S('command', 'beq', '$regA', '$regB', '$Lz'), \
S('command', 'j', '$Ly'), \
S('label', '$Lx'), \
remove_redundant_jumps(block)
self.assertEquals(block.statements, arguments)
def test_remove_redundant_jumps_bne_j_true(self):
block = B([self.foo,
S('command', 'bne', '$regA', '$regB', '$Lx'),
S('command', 'j', '$Ly'),
S('label', '$Lx'),
remove_redundant_jumps(block)
self.assertEquals(block.statements, [self.foo,
S('command', 'beq', '$regA', '$regB', '$Ly'),
S('label', '$Lx'),
def test_remove_redundant_jumps_bne_j_false(self):
arguments = [self.foo, \
S('command', 'bne', '$regA', '$regB', '$Lz'), \
S('command', 'j', '$Ly'), \
S('label', '$Lx'), \
remove_redundant_jumps(block)
self.assertEquals(block.statements, arguments)
def test_optimize_block_move_move_true(self):
block = B([self.foo,
S('command', 'move', '$regA', '$regB'),
S('command', 'move', '$regB', '$regA'),
optimize_block(block)
self.assertEquals(block.statements, [self.foo,
S('command', 'move', '$regA', '$regB'),
def test_optimize_block_mov_mov_false(self):
arguments = [self.foo, \
S('command', 'move', '$regA', '$regB'), \
S('command', 'move', '$regB', '$regC'), \
block = B(arguments)
optimize_block(block)
self.assertEquals(block.statements, arguments)