@@ -133,12 +133,14 @@ class Var(Variable):
133
133
var = Var ("v" , int )
134
134
self .assertEqual (var .typestring (), "Var" )
135
135
136
+
136
137
def test_copy (self ):
137
138
ip = Input ("ip" , float , norm )
138
139
self .assertTrue (ip .copy () is not ip )
139
140
self .assertEqual (ip .copy ().name , ip .name )
140
141
self .assertEqual (ip .copy ().datatype , ip .datatype )
141
142
self .assertEqual (ip .copy ().distribution , ip .distribution )
143
+ self .assertEqual (repr (ip ), repr (ip .copy ()))
142
144
143
145
144
146
class TestZ3Methods (unittest .TestCase ):
@@ -152,20 +154,41 @@ class TestZ3Methods(unittest.TestCase):
152
154
def setUp (self ) -> None :
153
155
self .i1 = Input ("i1" , int )
154
156
155
- def test_ge_add (self ):
156
- self .assertEqual (str (self .i1 + 1 >= 5 ), "i1 + 1 >= 5" )
157
+ def test_ge_self (self ):
158
+ self .assertEqual (str (self .i1 >= self .i1 ), "i1 >= i1" )
159
+
160
+ def test_add (self ):
161
+ self .assertEqual (str (self .i1 + 1 ), "i1 + 1" )
162
+
163
+ def test_ge (self ):
164
+ self .assertEqual (str (self .i1 >= 5 ), "i1 >= 5" )
165
+
166
+ def test_mod (self ):
167
+ self .assertEqual (str (self .i1 % 2 ), "i1%2" )
168
+
169
+ def test_ne (self ):
170
+ self .assertEqual (str (self .i1 != 5 ), "i1 != 5" )
171
+
172
+ def test_neg (self ):
173
+ self .assertEqual (str (- self .i1 ), "-i1" )
174
+
175
+ def test_pow (self ):
176
+ self .assertEqual (str (self .i1 ** 5 ), "i1**5" )
177
+
178
+ def test_le (self ):
179
+ self .assertEqual (str (self .i1 <= 5 ), "i1 <= 5" )
157
180
158
- def test_ne_mod (self ):
159
- self .assertEqual (str (self .i1 % 2 != 5 ), "i1%2 != 5 " )
181
+ def test_mul (self ):
182
+ self .assertEqual (str (self .i1 * 2 ), "i1*2 " )
160
183
161
- def test_neg_pow (self ):
162
- self .assertEqual (str (- self .i1 ** 5 ), "-i1** 5" )
184
+ def test_gt (self ):
185
+ self .assertEqual (str (self .i1 > 5 ), "i1 > 5" )
163
186
164
- def test_le_mul (self ):
165
- self .assertEqual (str (self .i1 * 2 <= 5 ), "i1*2 <= 5 " )
187
+ def test_truediv (self ):
188
+ self .assertEqual (str (self .i1 / 3 ), "i1/3 " )
166
189
167
- def test_gt_truediv (self ):
168
- self .assertEqual (str (self .i1 / 3 > 5 ), "i1/3 > 5 " )
190
+ def test_sub (self ):
191
+ self .assertEqual (str (self .i1 - 4 ), "i1 - 4 " )
169
192
170
- def test_lt_sub (self ):
171
- self .assertEqual (str (self .i1 - 4 < 5 ), "i1 - 4 < 5" )
193
+ def test_lt (self ):
194
+ self .assertEqual (str (self .i1 < 5 ), "i1 < 5" )
0 commit comments