test(*): fix warnings when running tests on Python 3.7

rubik · Sep 10, 2018 · 9ac4f7c · 9ac4f7c
1 parent b9f5ffd
commit 9ac4f7c
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/radon/tests/test_cli_harvest.py b/radon/tests/test_cli_harvest.py
@@ -1,4 +1,4 @@
-import collections
+import collections.abc
 
 import pytest
 
@@ -8,7 +8,7 @@
 
 
 BASE_CONFIG = Config(
-    exclude='test_[^.]+\.py',
+    exclude=r'test_[^.]+\.py',
     ignore='tests,docs',
 )
 
@@ -100,7 +100,7 @@ def test_base_to_terminal_not_implemented(base_config):
 def test_base_run(base_config):
     h = harvest.Harvester(['-'], base_config)
     h.gobble = fake_gobble
-    assert isinstance(h.run(), collections.Iterator)
+    assert isinstance(h.run(), collections.abc.Iterator)
     assert list(h.run()) == [('-', 42)]
     h.gobble = fake_gobble_raising
     assert list(h.run()) == [('-', {'error': 'mystr'})]
@@ -110,10 +110,10 @@ def test_base_results(base_config):
     h = harvest.Harvester([], base_config)
     h.run = fake_run
     results = h.results
-    assert isinstance(results, collections.Iterator)
+    assert isinstance(results, collections.abc.Iterator)
     assert list(results) == [{'file-0': 0}, {'file-1': 1}, {'file-2': 4}]
-    assert not isinstance(h.results, collections.Iterator)
-    assert isinstance(h.results, collections.Iterable)
+    assert not isinstance(h.results, collections.abc.Iterator)
+    assert isinstance(h.results, collections.abc.Iterable)
     assert isinstance(h.results, list)