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) |