1 import os
2 import sys
3 import unittest
4
5 from _testinternalcapi import perf_map_state_teardown, write_perf_map_entry
6
7 if sys.platform != 'linux':
8 raise unittest.SkipTest('Linux only')
9
10
11 class ESC[4;38;5;81mTestPerfMapWriting(ESC[4;38;5;149munittestESC[4;38;5;149m.ESC[4;38;5;149mTestCase):
12 def test_write_perf_map_entry(self):
13 self.assertEqual(write_perf_map_entry(0x1234, 5678, "entry1"), 0)
14 self.assertEqual(write_perf_map_entry(0x2345, 6789, "entry2"), 0)
15 with open(f"/tmp/perf-{os.getpid()}.map") as f:
16 perf_file_contents = f.read()
17 self.assertIn("1234 162e entry1", perf_file_contents)
18 self.assertIn("2345 1a85 entry2", perf_file_contents)
19 perf_map_state_teardown()