diff --git a/test/results/pexels.py b/test/results/pexels.py index 2a95f71a..f1052524 100644 --- a/test/results/pexels.py +++ b/test/results/pexels.py @@ -50,7 +50,10 @@ __tests__ = ( "following" : False, "hero" : bool, "id" : int, + "last_name" : {str, None}, + "location" : {str, None}, "slug" : str, + "username" : {str, None}, }, }, diff --git a/test/test_results.py b/test/test_results.py index f36f7984..c3b9b2de 100644 --- a/test/test_results.py +++ b/test/test_results.py @@ -237,7 +237,7 @@ class TestExtractorResults(unittest.TestCase): elif isinstance(test, range): self.assertRange(value, test, msg=path) elif isinstance(test, set): - self.assertIn(value, test, msg=path) + self.assertTrue(value in test or type(value) in test, msg=path) elif isinstance(test, list): subtest = False for idx, item in enumerate(test):