test functions in the order that they were declared

rudymatela · Aug 26, 2024 · b04d861 · b04d861
1 parent 94c9f3d
commit b04d861
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/TODO.md b/TODO.md
@@ -1,8 +1,6 @@
 TODO for leancheck.py
 =====================
 
-* test functions in the order that they were declared
-
 * add documentation to all functions
 
 * simplify code

diff --git a/src/leancheck.py b/src/leancheck.py
@@ -274,7 +274,12 @@ def holds(prop, max_tests=360):
 def testmod(max_tests=360, silent=False, verbose=False):
     n_failures = 0
     n_properties = 0
-    for name, member in getmembers(sys.modules["__main__"]):
+    def lineno(m):
+        try:
+            return m[1].__code__.co_firstlineno
+        except AttributeError:
+            return -1
+    for name, member in sorted(getmembers(sys.modules["__main__"]), key=lineno):
         if name.startswith("prop_") and callable(member):
             n_properties += 1
             passed = check(member, max_tests=max_tests, silent=silent, verbose=verbose)