提交 540e7d38 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Alternate strategy to not run the test in the base class.

上级 7d036b85
import numpy import numpy
import unittest import unittest
from nose.plugins.skip import SkipTest
import theano import theano
from theano import tensor from theano import tensor
from theano.tests import unittest_tools as utt from theano.tests import unittest_tools as utt
...@@ -56,7 +58,7 @@ class TestGetConvOutShape(unittest.TestCase): ...@@ -56,7 +58,7 @@ class TestGetConvOutShape(unittest.TestCase):
self.assertTrue(test4_params == (3, 4, 6, 4)) self.assertTrue(test4_params == (3, 4, 6, 4))
class BaseTestConv2d(object): class BaseTestConv2d(unittest.TestCase):
def setUp(self): def setUp(self):
self.inputs_shapes = [(8, 1, 12, 12), (8, 1, 18, 18), (2, 1, 4, 4), self.inputs_shapes = [(8, 1, 12, 12), (8, 1, 18, 18), (2, 1, 4, 4),
(6, 1, 10, 11), (2, 1, 6, 5), (1, 5, 9, 9)] (6, 1, 10, 11), (2, 1, 6, 5), (1, 5, 9, 9)]
...@@ -237,6 +239,8 @@ class BaseTestConv2d(object): ...@@ -237,6 +239,8 @@ class BaseTestConv2d(object):
mode=mode, eps=1) mode=mode, eps=1)
def test_all(self): def test_all(self):
if type(self) is BaseTestConv2d:
raise SkipTest("base class")
ds = [1, 1] ds = [1, 1]
db = (0, 0) db = (0, 0)
dflip = True in self.filter_flip dflip = True in self.filter_flip
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论