提交 dd3f7a56 authored 作者: James Bergstra's avatar James Bergstra 提交者: Frederic

TestOptimizationMixin raises proper SkipTest

上级 92d795e7
...@@ -4,6 +4,13 @@ import numpy ...@@ -4,6 +4,13 @@ import numpy
import theano.tensor as T import theano.tensor as T
from theano.configparser import config, AddConfigVar, StrParam from theano.configparser import config, AddConfigVar, StrParam
try:
from nose.plugins.skip import SkipTest
except ImportError:
class SkipTest(Exception):
"""
Skip this test
"""
AddConfigVar('unittests.rseed', AddConfigVar('unittests.rseed',
...@@ -93,7 +100,8 @@ class TestOptimizationMixin(object): ...@@ -93,7 +100,8 @@ class TestOptimizationMixin(object):
def assertFunctionContainsN(self, f, op, N): def assertFunctionContainsN(self, f, op, N):
return self.assertFunctionContains(f, op, min=N, max=N) return self.assertFunctionContains(f, op, min=N, max=N)
def SkipTest(self): def SkipTest(self, msg='Skip this test'):
raise Exception('how do I skip this test properly?') raise SkipTest(msg)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论