def test_enable_profile_all(self): lp = LineProfiler() lp.enable_profile_all() lp.enable() value = f(10) lp.disable() self.assertEqual(value, f(10)) self.assertEqual(len(lp.code_map.keys()), 1) self.assertEqual(len(lp.code_map[f.__code__]), 2)
from line_profiler import LineProfiler lp = LineProfiler() lp.enable_profile_all() profile = lp.runcall