From 84769e708c81a1600ff24e7140c889fefc4aa6ef Mon Sep 17 00:00:00 2001 From: Philipp Hagemeister Date: Thu, 13 Mar 2014 16:40:53 +0100 Subject: [ninegag] Fix extraction --- test/test_download.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'test/test_download.py') diff --git a/test/test_download.py b/test/test_download.py index bbbb6b78a..c241d9324 100644 --- a/test/test_download.py +++ b/test/test_download.py @@ -144,6 +144,10 @@ def generator(test_case): self.assertTrue( isinstance(got, compat_str) and match_rex.match(got), u'field %s (value: %r) should match %r' % (info_field, got, match_str)) + elif isinstance(expected, type): + got = info_dict.get(info_field) + self.assertTrue(isinstance(got, expected), + u'Expected type %r, but got value %r of type %r' % (expected, got, type(got))) else: if isinstance(expected, compat_str) and expected.startswith('md5:'): got = 'md5:' + md5(info_dict.get(info_field)) -- cgit 1.4.1