package/python-memray: new package

Signed-off-by: Fiona Klute (WIWA) <fiona.klute@gmx.de>
Signed-off-by: Thomas Petazzoni <thomas.petazzoni@bootlin.com>

support/testing/tests/package/test_python_memray.py

@@ -0,0 +1,38 @@
+from tests.package.test_python import TestPythonPackageBase
+import os
+
+
+class TestPythonMemray(TestPythonPackageBase):
+    __test__ = True
+    config = TestPythonPackageBase.config + \
+        """
+        BR2_PACKAGE_PYTHON3=y
+        BR2_PACKAGE_PYTHON_MEMRAY=y
+        """
+    sample_scripts = ["tests/package/sample_python_memray.py"]
+
+    def test_run(self):
+        self.login()
+        self.check_sample_scripts_exist()
+        test_trace = "/tmp/trace.bin"
+        test_flame = "/tmp/trace.html"
+
+        # profile the sample script
+        cmd = "%s -m memray run -o %s %s" % (self.interpreter, test_trace, os.path.basename(self.sample_scripts[0]))
+        output, exit_code = self.emulator.run(cmd, timeout=15)
+        self.assertEqual(exit_code, 0)
+        self.assertIn("Hello World!", output)
+
+        # generate statistics
+        cmd = "%s -m memray stats %s" % (self.interpreter, test_trace)
+        output, exit_code = self.emulator.run(cmd, timeout=15)
+        self.assertEqual(exit_code, 0)
+        self.assertTrue(any(filter(lambda line: "alloc" in line, output)))
+
+        # generate flamegraph
+        cmd = "%s -m memray flamegraph -o %s %s" % (self.interpreter, test_flame, test_trace)
+        output, exit_code = self.emulator.run(cmd, timeout=15)
+        self.assertEqual(exit_code, 0)
+        # rough plausibility check if the output file is HTML
+        cmd = "grep -i '</html>' %s" % (test_flame,)
+        self.assertEqual(exit_code, 0)