py2 and py3

getnikola · May 8, 2017 · fbd5730 · fbd5730
1 parent ce8682d
commit fbd5730
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_shortcodes.py b/tests/test_shortcodes.py
@@ -88,6 +88,9 @@ def test_errors(self):
 def test_extract_shortcodes(input, expected, monkeypatch):
 
     i = iter('SC%d' % i for i in range(1, 100))
-    monkeypatch.setattr(shortcodes, '_new_sc_id', i.next)
+    if sys.version[0] < 3:
+        monkeypatch.setattr(shortcodes, '_new_sc_id', i.next)
+    else:
+        monkeypatch.setattr(shortcodes, '_new_sc_id', i.__next__)
     extracted = shortcodes.extract_shortcodes(input)
     assert extracted == expected