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