From 16b65f76c376868d3c2ded357dfcd24f9175ffab Mon Sep 17 00:00:00 2001 From: Michael Lenzen Date: Fri, 30 Nov 2018 18:38:20 -0600 Subject: [PATCH] Fix test after import refactor --- tests/test_bags.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_bags.py b/tests/test_bags.py index 4d6d3c0..2b91298 100644 --- 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'))