Fix test after import refactor

mlenzen · Dec 1, 2018 · 16b65f7 · 16b65f7
1 parent 8f28aac
commit 16b65f7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_bags.py b/tests/test_bags.py
@@ -1,7 +1,8 @@
 """Test for bag classes."""
 import pytest
 
-from collections_extended.bags import _basebag, bag, frozenbag, _compat
+from collections_extended.bags import _basebag, bag, frozenbag
+from collections_extended._compat import is_py2
 
 
 def test_init():
@@ -38,7 +39,7 @@ def compare_bag_string(b):
 	assert "'c'" in str(_basebag('abracadabra'))
 	abra_elems = set(("'a'^5", "'b'^2", "'r'^2", "'c'", "'d'"))
 	assert compare_bag_string(bag('abracadabra')) == abra_elems
-	if not _compat.is_py2:
+	if not is_py2:
 		assert compare_bag_string(bag('abc')) == compare_bag_string(set('abc'))