helloworld · December 8, 2022 17:05 · Dec 8, 2022
diff --git a/modal_pytest.py b/modal_pytest.py
@@ -0,0 +1,55 @@
+import modal
+
+pytest_image = modal.Image.debian_slim().pip_install(["pytest"])
+stub = modal.Stub()
+
+code = """
+def hello_world(name=None, age=None, city=None):
+    return 'Hello World!'
+
+
+def test_hello_world():
+    assert hello_world() == 'Hello World!'
+
+
+def test_hello_world_with_name():
+    assert hello_world('John') == 'Hello John!'
+    assert hello_world(name='John') == 'Hello John!'
+
+
+def test_hello_world_with_name_and_age():
+    assert hello_world('John', 30) == 'Hello John! You are 30 years old.'
+    assert hello_world(
+        name='John', age=30) == 'Hello John! You are 30 years old.'
+
+
+def test_hello_world_with_name_and_age_and_city():
+    assert hello_world(
+        'John', 30, 'New York') == 'Hello John! You are 30 years old. You are from New York.'
+    assert hello_world(
+        name='John', age=30, city='New York') == 'Hello John! You are 30 years old. You are from New York.'
+    """
+
+
+@stub.function(image=pytest_image, shared_volumes={"/root/pytest": modal.SharedVolume()})
+def pytest_runner():
+    import tempfile
+    import subprocess
+
+    with tempfile.TemporaryDirectory() as dirname:
+        with open(f"{dirname}/test.py", "w") as f:
+            f.write(code)
+
+        try:
+            output = subprocess.check_output(
+                ["pytest", "test.py"], cwd=dirname, stderr=subprocess.STDOUT
+            ).decode("utf-8")
+        except subprocess.CalledProcessError as e:
+            output = e.output.decode("utf-8")
+
+        print(output)
+
+
+if __name__ == "__main__":
+    with stub.run():
+        pytest_runner()