Skip to content

Commit 122af2b

Browse files
committed
Fix categorical test
1 parent a4d40a3 commit 122af2b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/general/test_conversions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def test_categorical_object_domain_list():
6565
jdomain = feature_domain(domain)
6666
assert str(jdomain.getClass().getSimpleName()) == "ObjectFeatureDomain"
6767
assert jdomain.getCategories().size() == 2
68-
assert jdomain.getCategories().containsAll(domain)
68+
assert sorted([o.getObject() for o in jdomain.getCategories()]) == sorted(domain)
6969

7070

7171
def test_categorical_object_domain_list_2():
@@ -74,7 +74,7 @@ def test_categorical_object_domain_list_2():
7474
jdomain = feature_domain(domain)
7575
assert str(jdomain.getClass().getSimpleName()) == "ObjectFeatureDomain"
7676
assert jdomain.getCategories().size() == 2
77-
assert jdomain.getCategories().containsAll(domain)
77+
assert sorted([o.getObject() for o in jdomain.getCategories()]) == sorted(domain)
7878

7979

8080
def test_empty_domain():

0 commit comments

Comments
 (0)