提交 56bf2c46 authored 作者: Jeremiah Lowin's avatar Jeremiah Lowin

use cleaner representation of negative unit tests (per code review)

上级 e4830595
...@@ -5520,39 +5520,19 @@ class test_tensordot(unittest.TestCase): ...@@ -5520,39 +5520,19 @@ class test_tensordot(unittest.TestCase):
bvec = vector() bvec = vector()
# Test invalid length for axes # Test invalid length for axes
try: self.assertRaises(ValueError, tensordot, amat, bmat, (0, 1, 2))
c = tensordot(amat, bmat, (0, 1, 2))
assert False
except ValueError:
pass
# Test axes of uneven length # Test axes of uneven length
try: self.assertRaises(ValueError, tensordot, amat, bmat, ((0, 1), (0)))
c = tensordot(amat, bmat, ((0, 1), (0)))
assert False
except ValueError:
pass
# Test invalid len(axes) given inputs are matrices # Test invalid len(axes) given inputs are matrices
try: self.assertRaises(ValueError, tensordot, amat, bmat, ((0,1,2),(0,1,2)))
c = tensordot(amat, bmat, ((0, 1, 2), (0, 1, 2)))
assert False
except ValueError:
pass
# Test invalid axes[1] given that y is a vector # Test invalid axes[1] given that y is a vector
try: self.assertRaises(ValueError, tensordot, amat, bvec, (0, 1))
c = tensordot(amat, bvec, (0, 1))
assert False
except ValueError:
pass
# Test invalid scalar axes given inputs are matrices # Test invalid scalar axes given inputs are matrices
try: self.assertRaises(ValueError, tensordot, amat, bvec, 2)
c = tensordot(amat, bvec, 2)
assert False
except ValueError:
pass
def test_weird_valid_axes(self): def test_weird_valid_axes(self):
# Test matrix-matrix # Test matrix-matrix
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论