Use unittest.TestCase.assertIn

jupyter · Jun 30, 2017 · e52283a · e52283a
1 parent 6d068e8
commit e52283a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/nbviewer/providers/url/tests/test_url.py b/nbviewer/providers/url/tests/test_url.py
@@ -4,6 +4,7 @@
 #  Distributed under the terms of the BSD License.  The full license is in
 #  the file COPYING, distributed as part of this software.
 #-----------------------------------------------------------------------------
+import unittest
 
 import requests
 
@@ -13,7 +14,9 @@ class URLTestCase(NBViewerTestCase):
     def test_url(self):
         url = self.url('url/jdj.mit.edu/~stevenj/IJulia Preview.ipynb')
         r = requests.get(url)
-        self.assertIn(r.status_code, (200, 202))
+        # Base class overrides assertIn to do unicode in unicode checking
+        # We want to use the original unittest implementation
+        unittest.TestCase.assertIn(self, r.status_code, (200, 202))
         self.assertIn('Download Notebook', r.text)