Accept only my exception when testing Noparallel

This commit is contained in:
shortcutme 2019-12-17 16:06:13 +01:00
parent 909967629b
commit 87fc8ced5e
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE

View file

@ -132,15 +132,18 @@ class TestNoparallel:
assert 1.2 > taken >= 1.0 # 2 * 0.5s count = ~1s assert 1.2 > taken >= 1.0 # 2 * 0.5s count = ~1s
def testException(self, queue_spawn): def testException(self, queue_spawn):
class MyException(Exception):
pass
@util.Noparallel() @util.Noparallel()
def raiseException(): def raiseException():
raise Exception("Test error!") raise MyException("Test error!")
with pytest.raises(Exception) as err: with pytest.raises(MyException) as err:
raiseException() raiseException()
assert str(err.value) == "Test error!" assert str(err.value) == "Test error!"
with pytest.raises(Exception) as err: with pytest.raises(MyException) as err:
queue_spawn(raiseException).get() queue_spawn(raiseException).get()
assert str(err.value) == "Test error!" assert str(err.value) == "Test error!"