summaryrefslogtreecommitdiff
path: root/tools/python-yasm/tests/test_expr.py
blob: 97b021c3fc8793b6a9722b458179710d1e6b24cb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from tests import TestCase, add
from yasm import Expression
import operator

class TExpression(TestCase):
    def test_create(self):
        e1 = Expression(operator.add, 1, 2)
        e2 = Expression('+', 1, 2)
        
        self.assertEquals(e1.get_intnum(), e1.get_intnum())

    def test_extract(self):
        e1 = Expression('/', 15, 5)
        self.assertEquals(e1.get_intnum(), 3)
        self.assertRaises(ValueError, e1.extract_segoff)
        self.assertRaises(ValueError, e1.extract_wrt)

add(TExpression)