From 84769e708c81a1600ff24e7140c889fefc4aa6ef Mon Sep 17 00:00:00 2001
From: Philipp Hagemeister <phihag@phihag.de>
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')

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 v1.2.3