9
9
10
10
11
11
class TestDot :
12
+ def setup_method (self ):
13
+ numpy .random .seed (42 )
14
+
12
15
@pytest .mark .parametrize ("dtype" , get_all_dtypes ())
13
16
def test_dot_ones (self , dtype ):
14
17
n = 10 ** 5
@@ -355,6 +358,9 @@ def test_multi_dot(type):
355
358
356
359
357
360
class TestTensordot :
361
+ def setup_method (self ):
362
+ numpy .random .seed (87 )
363
+
358
364
@pytest .mark .parametrize ("dtype" , get_all_dtypes ())
359
365
def test_tensordot_scalar (self , dtype ):
360
366
a = 2
@@ -383,8 +389,7 @@ def test_tensordot(self, dtype, axes):
383
389
384
390
result = dpnp .tensordot (ia , ib , axes = axes )
385
391
expected = numpy .tensordot (a , b , axes = axes )
386
- # TODO: investigate the effect of factor, see SAT-6700
387
- assert_dtype_allclose (result , expected , factor = 24 )
392
+ assert_dtype_allclose (result , expected )
388
393
389
394
@pytest .mark .parametrize ("dtype" , get_complex_dtypes ())
390
395
@pytest .mark .parametrize ("axes" , [- 3 , - 2 , - 1 , 0 , 1 , 2 ])
@@ -400,8 +405,7 @@ def test_tensordot_complex(self, dtype, axes):
400
405
401
406
result = dpnp .tensordot (ia , ib , axes = axes )
402
407
expected = numpy .tensordot (a , b , axes = axes )
403
- # TODO: investigate the effect of factor, see SAT-6700
404
- assert_dtype_allclose (result , expected , factor = 24 )
408
+ assert_dtype_allclose (result , expected )
405
409
406
410
@pytest .mark .parametrize ("dtype" , get_all_dtypes (no_bool = True ))
407
411
@pytest .mark .parametrize (
@@ -426,8 +430,7 @@ def test_tensordot_axes(self, dtype, axes):
426
430
427
431
result = dpnp .tensordot (ia , ib , axes = axes )
428
432
expected = numpy .tensordot (a , b , axes = axes )
429
- # TODO: investigate the effect of factor, see SAT-6700
430
- assert_dtype_allclose (result , expected , factor = 24 )
433
+ assert_dtype_allclose (result , expected )
431
434
432
435
@pytest .mark .parametrize ("dtype1" , get_all_dtypes ())
433
436
@pytest .mark .parametrize ("dtype2" , get_all_dtypes ())
@@ -443,8 +446,7 @@ def test_tensordot_input_dtype_matrix(self, dtype1, dtype2):
443
446
444
447
result = dpnp .tensordot (ia , ib )
445
448
expected = numpy .tensordot (a , b )
446
- # TODO: investigate the effect of factor, see SAT-6700
447
- assert_dtype_allclose (result , expected , factor = 24 )
449
+ assert_dtype_allclose (result , expected )
448
450
449
451
def test_tensordot_strided (self ):
450
452
for dim in [1 , 2 , 3 , 4 ]:
@@ -500,6 +502,9 @@ def test_tensordot_error(self):
500
502
501
503
502
504
class TestVdot :
505
+ def setup_method (self ):
506
+ numpy .random .seed (42 )
507
+
503
508
@pytest .mark .parametrize ("dtype" , get_all_dtypes ())
504
509
def test_vdot_scalar (self , dtype ):
505
510
a = numpy .array ([3.5 ], dtype = dtype )
0 commit comments