mirror of
https://gitlab.com/buildroot.org/buildroot.git
synced 2026-08-01 21:23:51 -09:00
c2df8bab97
Signed-off-by: Fiona Klute (WIWA) <fiona.klute@gmx.de> Signed-off-by: Thomas Petazzoni <thomas.petazzoni@bootlin.com>
39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
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)
|