def test_toggle_cprofile_profiles(self): def my_method(): return 1 toggle_cprofile("my-process") my_method() toggle_cprofile("my-process") prof_path = self._get_prof_path("my-process") output = io.StringIO() stats = pstats.Stats(str(prof_path), stream=output) stats.print_callers() self.assertIn("my_method", output.getvalue())
def test_toggle_cprofile_writes_file(self): toggle_cprofile("my-process") toggle_cprofile("my-process") self.assertTrue(self._get_prof_path("my-process").exists())