提交 4171ff4c authored 作者: Nicolas Bouchard's avatar Nicolas Bouchard 提交者: Frederic

Fix name for nosetest

上级 84a1a252
......@@ -122,7 +122,10 @@ class TestCast(utt.InferShapeTester):
verify_grad_sparse(S2.Cast(t), [a], eps=eps)
class HVStackTester(utt.InferShapeTester):
class _HVStackTester(utt.InferShapeTester):
"""Test for both HStack and VStack.
"""
nb = 3 # Number of sparse matrix to stack
x = {}
mat = {}
......@@ -178,7 +181,14 @@ class HVStackTester(utt.InferShapeTester):
def _hv_switch(op, expected_function):
class XStackTester(HVStackTester):
"""Return the right test class for HStack or VStack.
:Parameters:
- `op`: HStack or VStack class.
- `expected_function`: function from scipy for comparaison.
"""
class XStackTester(_HVStackTester):
op_class = op
def expected_f(self, a, format=None, dtype=None):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论