From 50187818984af68f7443b24034c102c2e881daaf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mike=20F=C3=A4hrmann?= Date: Mon, 29 Apr 2019 17:27:59 +0200 Subject: [PATCH] allow type tests by name --- test/test_results.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/test/test_results.py b/test/test_results.py index fb786997..a8ed7ae0 100644 --- a/test/test_results.py +++ b/test/test_results.py @@ -119,8 +119,13 @@ class TestExtractorResults(unittest.TestCase): self._test_kwdict(value, test) elif isinstance(test, type): self.assertIsInstance(value, test, msg=key) - elif isinstance(test, str) and test.startswith("re:"): - self.assertRegex(value, test[3:], msg=key) + elif isinstance(test, str): + if test.startswith("re:"): + self.assertRegex(value, test[3:], msg=key) + elif test.startswith("type:"): + self.assertEqual(type(value).__name__, test[5:], msg=key) + else: + self.assertEqual(value, test, msg=key) else: self.assertEqual(value, test, msg=key)