1from tests import TestCase, add 2from yasm import Expression 3import operator 4 5class TExpression(TestCase): 6 def test_create(self): 7 e1 = Expression(operator.add, 1, 2) 8 e2 = Expression('+', 1, 2) 9 10 self.assertEquals(e1.get_intnum(), e1.get_intnum()) 11 12 def test_extract(self): 13 e1 = Expression('/', 15, 5) 14 self.assertEquals(e1.get_intnum(), 3) 15 self.assertRaises(ValueError, e1.extract_segoff) 16 self.assertRaises(ValueError, e1.extract_wrt) 17 18add(TExpression) 19