tests pass

[nmutil.git] / src / nmutil / test / __init__.py

diff --git a/src/nmutil/test/__init__.py b/src/nmutil/test/__init__.py
index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0a3071d00d0865bb7963278d33135e1169c171d3 100644 (file)
--- a/src/nmutil/test/__init__.py
+++ b/src/nmutil/test/__init__.py
@@ -0,0 +1,16 @@
+class StepLimiter:
+    def __init__(self, limit=100000):
+        self.limit = limit
+        self.step_count = 0
+        assert self.step_count < self.limit, "step count limit reached"
+
+    def step(self):
+        self.step_count += 1
+        assert self.step_count < self.limit, "step count limit reached"
+
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        self.step()
+        return None