Skip to content

Commit 1c52a41

Browse files
create classmethod for reseting
1 parent 9b7bb19 commit 1c52a41

File tree

2 files changed

+9
-5
lines changed

2 files changed

+9
-5
lines changed

tests/test_utils.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,12 +14,10 @@
1414

1515

1616
@pytest.fixture(autouse=True)
17-
def clear_seq_state():
18-
seq._instances = {}
19-
seq._locks = {}
17+
def reset_seq():
18+
seq._reset()
2019
yield
21-
seq._instances = {}
22-
seq._locks = {}
20+
seq._reset()
2321

2422

2523
def test_basic_number_sequence():

tests/utils.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,12 @@ def _generate_text_value(self) -> str:
209209
stringified_value = [str(v) for v in value]
210210
return "".join(stringified_value)
211211

212+
@classmethod
213+
def _reset(cls):
214+
"""Reset all sequence state. Used for testing purposes."""
215+
cls._instances.clear()
216+
cls._locks.clear()
217+
212218
@classmethod
213219
def iter(
214220
cls,

0 commit comments

Comments
 (0)