build pytest suite out of test cases

Author: ajmirsky

pyproject.toml

@@ -20,13 +20,15 @@ classifiers = [
 ]
 requires-python = ">=3.9"
 dependencies = [
-    "psutil>=0.0"
+    "psutil>=0.0",
 ]
 
 [dependency-groups]
 dev = [
-    "hatch>=1.14.1"
-    ]
+    "hatch>=1.14.1",
+    "pytest>=8.4.1",
+    "pytest-asyncio>=1.0.0",
+]
 
 [project.scripts]
 mprof = "memory_profiler.mprof:main"
@@ -58,3 +60,6 @@ source = "vcs"
 
 [tool.codespell]
 skip = './.git'
+
+[tool.pytest.ini_options]
+testpaths = ["test"]

src/memory_profiler/__init__.py

@@ -0,0 +1,9 @@
+from .__main__ import (profile, memory_usage, __version__, LineProfiler, show_results)
+
+__all__ = [
+    'profile',
+    'memory_usage',
+    '__version__',
+    'LineProfiler',
+    'show_results'
+]

test/test_as.py

@@ -1,9 +1,10 @@
 import math
+from memory_profiler import profile
 
 @profile
 def f():
     o = math.sqrt(2013)
     return o
 
-if __name__ == '__main__':
+def test_f():
     f()

test/test_async.py

@@ -1,5 +1,6 @@
 import asyncio
-import sys
+
+import pytest
 
 from memory_profiler import profile
 
@@ -10,18 +11,13 @@ async def my_func():
     b = [2] * (2 * 10 ** 7)
     await asyncio.sleep(1e-2)
     del b
+    return 1, 2
 
-async def main():
+@pytest.mark.asyncio
+async def test_async_func():
     task = asyncio.create_task(my_func())
     res = await asyncio.gather(task)
+    assert res, "function didn't return anything"
+    assert len(res) > 0, "task didn't return anything"
+    assert res == [(1, 2), ]
 
-async def main_legacy():
-    future = asyncio.ensure_future(my_func())
-    res = await asyncio.gather(future)
-
-if __name__ == '__main__':
-    if sys.version_info >= (3, 7):
-        asyncio.run(main())  # main loop
-    else:
-        loop = asyncio.get_event_loop()
-        loop.run_until_complete(main_legacy())

test/test_attributes.py

@@ -2,11 +2,11 @@
 
 
 @profile
-def test_with_profile(arg1):
+def func_with_profile(arg1):
     """dummy doc"""
     return None
 
 
-if __name__ == '__main__':
-    assert test_with_profile.__doc__ == "dummy doc"
-    assert test_with_profile.__name__ == "test_with_profile"
+def test_with_profile():
+    assert func_with_profile.__doc__ == "dummy doc"
+    assert func_with_profile.__name__ == "func_with_profile"

test/test_exception.py

@@ -1,12 +1,12 @@
+import pytest
 
 # make sure that memory profiler does not hang on exception
 from memory_profiler import memory_usage
 
 def foo():
     raise NotImplementedError('Error')
 
-try:
-    out = memory_usage((foo, tuple(), {}), timeout=1)
-except NotImplementedError:
-    pass
-print('Success')
+def test_hang_on_exception():
+
+    with pytest.raises(NotImplementedError):
+        out = memory_usage((foo, tuple(), {}), timeout=1)

test/test_exit_code.py

@@ -8,7 +8,7 @@ class TestExitCode(unittest.TestCase):
     def setUp(self):
         # to be able to import mprof
         sys.path.append('.')
-        from mprof import run_action
+        from memory_profiler.mprof import run_action
         self.run_action = run_action
 
     def test_exit_code_success(self):

test/test_func.py

@@ -1,11 +1,13 @@
+from memory_profiler import profile
 
 @profile
-def test_1(i):
+def func_1(i):
     # .. will be called twice ..
     c = {}
     for i in range(i):
         c[i] = 2
 
-if __name__ == '__main__':
-    test_1(10)
-    test_1(10000)
+def test_func():
+
+    func_1(10)
+    func_1(10000)

test/test_gen.py

@@ -1,3 +1,4 @@
+from memory_profiler import profile
 
 @profile
 def my_func():
@@ -8,7 +9,7 @@ def my_func():
 
 
 @profile
-def test_comprehension():
+def func_comprehension():
     # Dict comprehension
     d_comp = dict((str(k*k), [v] * (1<<17))
                   for (v, k) in enumerate(range(99, 111)))
@@ -33,7 +34,7 @@ def hh(x=1):
 
 
 @profile
-def test_generator():
+def func_generator():
     a_gen = ([42] * (1<<20) for __ in '123')
     huge_lst = list(a_gen)
 
@@ -43,10 +44,14 @@ def test_generator():
 
     return a_gen
 
+def test_gen():
+
+    # with profile:
+    next(my_func())     # Issue #42
+
+def test_generator():
+    _ = func_generator()
 
-if __name__ == '__main__':
-    with profile:
-        next(my_func())     # Issue #42
 
-    test_generator()
-    test_comprehension()
+def test_comprehension():
+    _ = func_comprehension()

test/test_import.py

@@ -8,5 +8,5 @@ def my_func():
     del b
     return a
 
-if __name__ == '__main__':
+def test_my_func():
     my_func()