[python_basic_memory] Initial commit

Author: KowalskiThomas

scenarios/python_basic_memory_3.11/Dockerfile

@@ -0,0 +1,20 @@
+ARG BASE_IMAGE="prof-python-3.11"
+FROM $BASE_IMAGE
+
+# Copy the Python target into the container
+COPY ./scenarios/python_basic_memory_3.11/requirements.txt /app/requirements.txt
+RUN chmod 644 /app/*
+
+RUN pip install -r /app/requirements.txt
+
+COPY ./scenarios/python_basic_memory_3.11/main.py /app/main.py
+
+# Set the working directory to the location of the program
+WORKDIR /app
+
+ENV EXECUTION_TIME_SEC="2"
+ENV DD_TRACE_DEBUG=true
+
+# Run the program when the container starts
+CMD python main.py
+# CMD ddprof -l notice --preset cpu_live_heap python main.py

scenarios/python_basic_memory_3.11/expected_profile.json

@@ -0,0 +1,41 @@
+{
+  "test_name": "python_basic",
+  "pprof-regex": "",
+  "stacks": [
+    {
+      "profile-type": "wall-time",
+      "pprof-regex": "",
+      "stack-content": [
+        {
+          "regular_expression": "^\u003cmodule\u003e;target$",
+          "value": 1500000000,
+          "error_margin": 20,
+          "labels": [
+            {
+              "key": "thread name",
+              "values": [
+                "MainThread"
+              ],
+              "values_regex": ""
+            }
+          ]
+        },
+        {
+          "regular_expression": ".*run;target$",
+          "value": 500000000,
+          "error_margin": 25,
+          "labels": [
+            {
+              "key": "thread name",
+              "values": [
+                "Thread-1 (target)"
+              ],
+              "values_regex": ""
+            }
+          ]
+        }
+      ]
+    }
+  ],
+  "scale_by_duration": false
+}

scenarios/python_basic_memory_3.11/main.py

@@ -0,0 +1,29 @@
+from datetime import time
+import os
+import time
+from ddtrace.profiling import Profiler
+
+
+class Target:
+    def __init__(self):
+        self.memory = []
+
+    def run(self, n):
+        end_time = time.monotonic() + n
+        while time.monotonic() < end_time:
+            self.memory.append(bytearray(1024))
+
+
+if __name__ == "__main__":
+    # Simple application that creates two threads with different durations:
+    # - MainThread runs target() for 2 seconds
+    # - Worker Thread-1 runs target() for 1 second
+    # The profiler should capture both threads with their respective durations.
+    prof = Profiler()
+    prof.start()  # Should be as early as possible, eg before other imports, to ensure everything is profiled
+
+    EXECUTION_TIME_SEC = int(os.environ.get("EXECUTION_TIME_SEC", "2"))
+
+    Target().run(EXECUTION_TIME_SEC)
+
+    prof.stop()

scenarios/python_basic_memory_3.11/requirements.txt

@@ -0,0 +1 @@
+ddtrace==4.0.0