提交 e063f64b authored 作者: Frederic Bastien's avatar Frederic Bastien

small test clean up

上级 107c28d1
...@@ -310,8 +310,6 @@ class T_random_function(utt.InferShapeTester): ...@@ -310,8 +310,6 @@ class T_random_function(utt.InferShapeTester):
self.assertRaises(ValueError, fm11) self.assertRaises(ValueError, fm11)
self.assertRaises(ValueError, fm12) self.assertRaises(ValueError, fm12)
u01, u02 = f0() u01, u02 = f0()
print(u01)
print(u02)
self.assertTrue(numpy.allclose(u01, u02[0])) self.assertTrue(numpy.allclose(u01, u02[0]))
def test_uniform(self): def test_uniform(self):
...@@ -332,10 +330,6 @@ class T_random_function(utt.InferShapeTester): ...@@ -332,10 +330,6 @@ class T_random_function(utt.InferShapeTester):
val1 = f() val1 = f()
numpy_val0 = numpy_rng.uniform(-2.0, 2.0, size=(4,)) numpy_val0 = numpy_rng.uniform(-2.0, 2.0, size=(4,))
numpy_val1 = numpy_rng.uniform(-2.0, 2.0, size=(4,)) numpy_val1 = numpy_rng.uniform(-2.0, 2.0, size=(4,))
print(val0)
print(numpy_val0)
print(val1)
print(numpy_val1)
self.assertTrue(numpy.allclose(val0, numpy_val0)) self.assertTrue(numpy.allclose(val0, numpy_val0))
self.assertTrue(numpy.allclose(val1, numpy_val1)) self.assertTrue(numpy.allclose(val1, numpy_val1))
...@@ -359,10 +353,6 @@ class T_random_function(utt.InferShapeTester): ...@@ -359,10 +353,6 @@ class T_random_function(utt.InferShapeTester):
val1 = f() val1 = f()
numpy_val0 = numpy_rng.binomial(5, 0.8, size=(7, 12)) numpy_val0 = numpy_rng.binomial(5, 0.8, size=(7, 12))
numpy_val1 = numpy_rng.binomial(5, 0.8, size=(7, 12)) numpy_val1 = numpy_rng.binomial(5, 0.8, size=(7, 12))
print(val0)
print(numpy_val0)
print(val1)
print(numpy_val1)
self.assertTrue(numpy.all(val0 == numpy_val0)) self.assertTrue(numpy.all(val0 == numpy_val0))
self.assertTrue(numpy.all(val1 == numpy_val1)) self.assertTrue(numpy.all(val1 == numpy_val1))
...@@ -384,10 +374,6 @@ class T_random_function(utt.InferShapeTester): ...@@ -384,10 +374,6 @@ class T_random_function(utt.InferShapeTester):
val1 = f() val1 = f()
numpy_val0 = numpy_rng.normal(4.0, 2.0, size=(2, 3)) numpy_val0 = numpy_rng.normal(4.0, 2.0, size=(2, 3))
numpy_val1 = numpy_rng.normal(4.0, 2.0, size=(2, 3)) numpy_val1 = numpy_rng.normal(4.0, 2.0, size=(2, 3))
print(val0)
print(numpy_val0)
print(val1)
print(numpy_val1)
self.assertTrue(numpy.allclose(val0, numpy_val0)) self.assertTrue(numpy.allclose(val0, numpy_val0))
self.assertTrue(numpy.allclose(val1, numpy_val1)) self.assertTrue(numpy.allclose(val1, numpy_val1))
...@@ -411,10 +397,6 @@ class T_random_function(utt.InferShapeTester): ...@@ -411,10 +397,6 @@ class T_random_function(utt.InferShapeTester):
val1 = f() val1 = f()
numpy_val0 = numpy_rng.random_integers(-3, 16, size=(11, 8)) numpy_val0 = numpy_rng.random_integers(-3, 16, size=(11, 8))
numpy_val1 = numpy_rng.random_integers(-3, 16, size=(11, 8)) numpy_val1 = numpy_rng.random_integers(-3, 16, size=(11, 8))
print(val0)
print(numpy_val0)
print(val1)
print(numpy_val1)
self.assertTrue(numpy.allclose(val0, numpy_val0)) self.assertTrue(numpy.allclose(val0, numpy_val0))
self.assertTrue(numpy.allclose(val1, numpy_val1)) self.assertTrue(numpy.allclose(val1, numpy_val1))
...@@ -447,10 +429,6 @@ class T_random_function(utt.InferShapeTester): ...@@ -447,10 +429,6 @@ class T_random_function(utt.InferShapeTester):
for i in range(7)]) for i in range(7)])
numpy_val1 = numpy.asarray([numpy_rng.permutation(8) numpy_val1 = numpy.asarray([numpy_rng.permutation(8)
for i in range(7)]) for i in range(7)])
print(val0)
print(numpy_val0)
print(val1)
print(numpy_val1)
self.assertTrue(numpy.all(val0 == numpy_val0)) self.assertTrue(numpy.all(val0 == numpy_val0))
self.assertTrue(numpy.all(val1 == numpy_val1)) self.assertTrue(numpy.all(val1 == numpy_val1))
...@@ -502,10 +480,6 @@ class T_random_function(utt.InferShapeTester): ...@@ -502,10 +480,6 @@ class T_random_function(utt.InferShapeTester):
val1 = f() val1 = f()
numpy_val0 = numpy_rng.choice(10, (11, 8), True, None) numpy_val0 = numpy_rng.choice(10, (11, 8), True, None)
numpy_val1 = numpy_rng.choice(10, (11, 8), True, None) numpy_val1 = numpy_rng.choice(10, (11, 8), True, None)
print(val0)
print(numpy_val0)
print(val1)
print(numpy_val1)
self.assertTrue(numpy.allclose(val0, numpy_val0)) self.assertTrue(numpy.allclose(val0, numpy_val0))
self.assertTrue(numpy.allclose(val1, numpy_val1)) self.assertTrue(numpy.allclose(val1, numpy_val1))
...@@ -529,10 +503,6 @@ class T_random_function(utt.InferShapeTester): ...@@ -529,10 +503,6 @@ class T_random_function(utt.InferShapeTester):
val1 = f() val1 = f()
numpy_val0 = numpy_rng.poisson(5, size=(11, 8)) numpy_val0 = numpy_rng.poisson(5, size=(11, 8))
numpy_val1 = numpy_rng.poisson(5, size=(11, 8)) numpy_val1 = numpy_rng.poisson(5, size=(11, 8))
print(val0)
print(numpy_val0)
print(val1)
print(numpy_val1)
self.assertTrue(numpy.allclose(val0, numpy_val0)) self.assertTrue(numpy.allclose(val0, numpy_val0))
self.assertTrue(numpy.allclose(val1, numpy_val1)) self.assertTrue(numpy.allclose(val1, numpy_val1))
...@@ -541,7 +511,6 @@ class T_random_function(utt.InferShapeTester): ...@@ -541,7 +511,6 @@ class T_random_function(utt.InferShapeTester):
results as numpy.""" results as numpy."""
rng_R = random_state_type() rng_R = random_state_type()
post_r, out = permutation(rng_R, size=(9,), n=6) post_r, out = permutation(rng_R, size=(9,), n=6)
print('OUT NDIM', out.ndim)
f = compile.function( f = compile.function(
[compile.In(rng_R, [compile.In(rng_R,
value=numpy.random.RandomState(utt.fetch_seed()), value=numpy.random.RandomState(utt.fetch_seed()),
...@@ -558,10 +527,6 @@ class T_random_function(utt.InferShapeTester): ...@@ -558,10 +527,6 @@ class T_random_function(utt.InferShapeTester):
for i in range(9)]) for i in range(9)])
numpy_val1 = numpy.asarray([numpy_rng.permutation(6) numpy_val1 = numpy.asarray([numpy_rng.permutation(6)
for i in range(9)]) for i in range(9)])
print(val0)
print(numpy_val0)
print(val1)
print(numpy_val1)
self.assertTrue(numpy.all(val0 == numpy_val0)) self.assertTrue(numpy.all(val0 == numpy_val0))
self.assertTrue(numpy.all(val1 == numpy_val1)) self.assertTrue(numpy.all(val1 == numpy_val1))
...@@ -596,10 +561,6 @@ class T_random_function(utt.InferShapeTester): ...@@ -596,10 +561,6 @@ class T_random_function(utt.InferShapeTester):
val1, = f() val1, = f()
numpy_val0 = numpy_rng.multinomial(6, [0.2] * 5, (7, 3)) numpy_val0 = numpy_rng.multinomial(6, [0.2] * 5, (7, 3))
numpy_val1 = numpy_rng.multinomial(6, [0.2] * 5, (7, 3)) numpy_val1 = numpy_rng.multinomial(6, [0.2] * 5, (7, 3))
print(val0)
print(numpy_val0)
print(val1)
print(numpy_val1)
self.assertTrue(numpy.all(val0 == numpy_val0)) self.assertTrue(numpy.all(val0 == numpy_val0))
self.assertTrue(numpy.all(val1 == numpy_val1)) self.assertTrue(numpy.all(val1 == numpy_val1))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论