sympycore/tests/test_lowlevel.py

   1
   2 from sympycore.core import Expr, Pair, heads, IntegerList
   3 from sympycore.heads import *
   4
   5 class MyExpr(Expr):
   6
   7     @classmethod
   8     def convert(cls, obj, typeerror=True):
   9         if isinstance(obj, cls):
  10             return obj
  11         if isinstance(obj, (int, long, float, complex)):
  12             return MyExpr(NUMBER, obj)
  13         if isinstance(obj, str):
  14             return MyExpr(SYMBOL, obj)
  15         if typeerror:
  16             raise TypeError('Cannot convert type %r to %r' % (type(obj), cls.__name__))
  17         return NotImplemented
  18
  19     def __add__(self, other):
  20         return Add(self, self.convert(other))
  21
  22     __radd__ = __add__
  23
  24 class AExpr(Expr):
  25
  26     pass
  27
  28 def Number(n):
  29     return MyExpr(NUMBER, n)
  30 def Symbol(s):
  31     return MyExpr(SYMBOL, s)
  32 def Add(x, y):
  33     d = {}
  34     if x==y:
  35         d[x] = 2
  36     else:
  37         d[x] = 1
  38         d[y] = 1
  39     return MyExpr(TERMS, d)
  40
  41 def test_equality_number():
  42     assert Number(1)==Number(1)
  43     assert Number(1)==Number(1L)
  44     assert Number(1)==1
  45     assert Number(1)==1.0
  46     assert Number(1.0)==1
  47     assert Number(1)==1.0+0j
  48     assert Number(1)==1L
  49     assert 1==Number(1)
  50     assert 1L==Number(1)
  51     assert 1==Number(1L)
  52
  53     assert Number(1)!=Number(2)
  54     assert Number(1)!=Number(2L)
  55     assert Number(1)!=2
  56     assert Number(1)!=2L
  57     assert 2!=Number(1)
  58
  59 def test_equality_symbol():
  60     assert Symbol('x')==Symbol('x')
  61     assert Symbol('x')=='x'
  62     assert 'x'==Symbol('x')
  63
  64     assert Symbol('x')!=Symbol('y')
  65     assert Symbol('x')!='y'
  66     assert 'x'!=Symbol('y')
  67
  68     assert MyExpr(SYMBOL, 'x')==AExpr(SYMBOL, 'x')
  69
  70 def test_pos():
  71     x, y, z = map(Symbol,'xyz')
  72     assert MyExpr(POS,x)==x
  73     assert hash(MyExpr(POS,x))==hash(x)
  74
  75 def test_neg():
  76     x, y, z = map(Symbol,'xyz')
  77     z, o, t = map(Number, [0,1,2])
  78     assert MyExpr(NEG, t)==-2, MyExpr(NEG, t)
  79     assert hash(MyExpr(NEG, t))==hash(-2), `hash(MyExpr(NEG, t)),hash(-2)`
  80
  81 def test_equality_add():
  82     x, y, z = map(Symbol,'xyz')
  83     assert x + y == y + x
  84     assert x + x == x + x
  85     assert x + y == (TERMS, {x:1, y:1})
  86
  87     assert x + y != y + z
  88     assert x + y != x
  89     assert x + y != 'x'
  90     assert x + y != 1
  91
  92     assert MyExpr(ADD,[])==0,`MyExpr(ADD,[]).as_lowlevel()`
  93     assert MyExpr(ADD,[x])==x
  94     assert (not not MyExpr(ADD,[])) == False
  95
  96     assert hash(MyExpr(ADD,[]))==hash(0)
  97     assert hash(MyExpr(ADD,[x]))==hash(x)
  98
  99     x1 = AExpr(SYMBOL, 'x')
 100     y1 = AExpr(SYMBOL, 'y')
 101
 102     assert AExpr(ADD,[x1])==MyExpr(ADD,[x])
 103     assert AExpr(ADD,[x1,y1])==MyExpr(ADD,[x,y])
 104     assert not (AExpr(ADD,[x1,y1])<MyExpr(ADD,[x,y]))
 105     assert not (AExpr(ADD,[x1,y1])>MyExpr(ADD,[x,y]))
 106
 107     assert (AExpr(ADD,[x1,y1])<=MyExpr(ADD,[x,y]))
 108     assert (AExpr(ADD,[x1,y1])>=MyExpr(ADD,[x,y]))
 109
 110 def test_equality_mul():
 111     x, y, z = map(Symbol,'xyz')
 112     assert (not not MyExpr(MUL,[])) == True
 113     assert MyExpr(MUL,[])==1
 114     assert MyExpr(MUL,[x])==x
 115     assert hash(MyExpr(MUL,[]))==hash(1)
 116     assert hash(MyExpr(MUL,[x]))==hash(x)
 117
 118 def test_equality_term_coeff():
 119     x, y, z = map(Symbol,'xyz')
 120     z, o, t = map(Number, [0,1,2])
 121     assert MyExpr(TERM_COEFF, (x, 0))==0
 122     assert MyExpr(TERM_COEFF, (x, 1))==x
 123     assert MyExpr(TERM_COEFF, (o, 2))==2
 124     assert MyExpr(TERM_COEFF, (x, -1))==MyExpr(NEG, x)
 125     assert hash(MyExpr(TERM_COEFF, (x, 0)))==hash(0)
 126     assert hash(MyExpr(TERM_COEFF, (x, 1)))==hash(x)
 127     assert hash(MyExpr(TERM_COEFF, (o, 2)))==hash(2)
 128     assert hash(MyExpr(TERM_COEFF, (x, -1)))==hash(MyExpr(NEG, x))
 129
 130
 131 def test_equality_term_coeff_dict():
 132     x, y, z = map(Symbol,'xyz')
 133     z, o, t = map(Number, [0,1,2])
 134     assert MyExpr(TERM_COEFF_DICT, {})==0
 135     assert MyExpr(TERM_COEFF_DICT, {x:0})==0
 136     assert MyExpr(TERM_COEFF_DICT, {x:1})==x
 137     assert MyExpr(TERM_COEFF_DICT, {x:2})==MyExpr(TERM_COEFF, (x, 2))
 138     assert MyExpr(TERM_COEFF_DICT, {o:2})==2
 139
 140     assert hash(MyExpr(TERM_COEFF_DICT, {}))==hash(0)
 141     assert hash(MyExpr(TERM_COEFF_DICT, {x:0}))==hash(0)
 142     assert hash(MyExpr(TERM_COEFF_DICT, {x:1}))==hash(x)
 143     assert hash(MyExpr(TERM_COEFF_DICT, {x:2}))==hash(MyExpr(TERM_COEFF, (x, 2)))
 144     assert hash(MyExpr(TERM_COEFF_DICT, {o:2}))==hash(2)
 145
 146 def test_equality_pow():
 147     x, y, z = map(Symbol,'xyz')
 148     z, o = map(Number, [0,1])
 149     assert MyExpr(POW, (x, 0))==1
 150     assert MyExpr(POW, (x, 1))==x
 151     assert MyExpr(POW, (o, x))==1
 152
 153     assert hash(MyExpr(POW, (x, 0)))==hash(1)
 154     assert hash(MyExpr(POW, (x, 1)))==hash(x)
 155     assert hash(MyExpr(POW, (o, x)))==hash(1)
 156
 157 def test_equality_base_exp_dict():
 158     x, y, z = map(Symbol,'xyz')
 159     z, o, t = map(Number, [0,1,2])
 160     assert MyExpr(BASE_EXP_DICT, {})==1
 161     assert MyExpr(BASE_EXP_DICT, {x:0})==1
 162     assert MyExpr(BASE_EXP_DICT, {x:1})==x
 163     assert MyExpr(BASE_EXP_DICT, {x:2})==MyExpr(POW, (x, 2))
 164     assert MyExpr(BASE_EXP_DICT, {o:2})==1
 165
 166     assert hash(MyExpr(BASE_EXP_DICT, {}))==hash(1)
 167     assert hash(MyExpr(BASE_EXP_DICT, {x:0}))==hash(1)
 168     assert hash(MyExpr(BASE_EXP_DICT, {x:1}))==hash(x)
 169     assert hash(MyExpr(BASE_EXP_DICT, {x:2}))==hash(MyExpr(POW, (x, 2)))
 170     assert hash(MyExpr(BASE_EXP_DICT, {o:2}))==hash(1)
 171
 172 def test_equality_exp_coeff_dict():
 173     x, y, z = map(Symbol,'xyz')
 174     assert MyExpr(EXP_COEFF_DICT, Pair((x,), {}))==0
 175     assert MyExpr(EXP_COEFF_DICT, Pair((x,), {IntegerList(0):2}))==2
 176     assert MyExpr(EXP_COEFF_DICT, Pair((x,), {IntegerList(1):1}))==x
 177     assert MyExpr(EXP_COEFF_DICT, Pair((x,), {IntegerList(1):2}))==MyExpr(TERM_COEFF, (x,2))
 178     assert MyExpr(EXP_COEFF_DICT, Pair((x,), {IntegerList(2):1}))==MyExpr(POW, (x,2))
 179
 180 def test_hash_number():
 181     assert hash(Number(1))==hash(1)
 182     assert hash(Number(-1))==hash(-1)
 183     assert hash(Number(1212424))==hash(1212424)
 184     assert hash(Number(-1212424))==hash(-1212424)
 185
 186 def test_hash_symbol():
 187     assert hash(Symbol('x'))==hash('x'),`hash(Symbol('x')),hash('x')`
 188     assert hash(Symbol('y'))==hash('y')
 189
 190 def test_hash_dict_data():
 191     x, y, z = map(Symbol,'xyz')
 192     assert hash(x + y) == hash((TERMS, frozenset([(x,1),(y,1)])))
 193
 194 def test_hash_list_data():
 195     l = map(MyExpr,[1,2,3])
 196     e1 = MyExpr(MUL, l)
 197     assert e1.is_writable
 198     e2 = MyExpr(MUL, tuple(l))
 199     assert hash(e1)==hash(e2)
 200     assert not e1.is_writable
 201
 202 def test_is_writable():
 203     n, m = map(MyExpr, [1,2])
 204     assert MyExpr(MUL, [n,m]).is_writable
 205     assert not MyExpr(MUL, (n,m)).is_writable
 206