Skip to content

Commit f2c8525

Browse files
revert subset test update
1 parent 6e534c6 commit f2c8525

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

ote_sdk/ote_sdk/tests/entities/test_subset.py

+6-7
Original file line numberDiff line numberDiff line change
@@ -43,22 +43,21 @@ def test_subset_members(self):
4343
TESTING = 3
4444
UNLABELED = 4
4545
PSEUDOLABELED = 5
46-
UNASSIGNED = 6
4746
4847
<b>Steps</b>
4948
1. Create enum instance
5049
2. Check members
5150
"""
5251
test_instance = Subset
5352

54-
for i in range(0, 7):
53+
for i in range(0, 6):
5554
assert test_instance(i) in list(Subset)
5655

5756
with pytest.raises(AttributeError):
5857
test_instance.WRONG
5958

6059
with pytest.raises(ValueError):
61-
test_instance(7)
60+
test_instance(6)
6261

6362
@pytest.mark.priority_medium
6463
@pytest.mark.unit
@@ -82,14 +81,14 @@ def test_subset_magic_str(self):
8281
test_instance = Subset
8382
magic_str_list = [str(i) for i in list(Subset)]
8483

85-
for i in range(0, 7):
84+
for i in range(0, 6):
8685
assert str(test_instance(i)) in magic_str_list
8786

8887
with pytest.raises(AttributeError):
8988
str(test_instance.WRONG)
9089

9190
with pytest.raises(ValueError):
92-
str(test_instance(7))
91+
str(test_instance(6))
9392

9493
assert len(set(magic_str_list)) == len(magic_str_list)
9594

@@ -114,13 +113,13 @@ def test_subset_magic_repr(self):
114113
test_instance = Subset
115114
magic_repr_list = [repr(i) for i in list(Subset)]
116115

117-
for i in range(0, 7):
116+
for i in range(0, 6):
118117
assert repr(test_instance(i)) in magic_repr_list
119118

120119
with pytest.raises(AttributeError):
121120
repr(test_instance.WRONG)
122121

123122
with pytest.raises(ValueError):
124-
repr(test_instance(7))
123+
repr(test_instance(6))
125124

126125
assert len(set(magic_repr_list)) == len(magic_repr_list)

0 commit comments

Comments
 (0)