def run(name, trace_filter=None, config=None, comment=None):
if not config:
config = Config()
if trace_filter:
config.trace_filter = trace_filter
graphviz = GraphvizOutput()
graphviz.output_file = 'filter-{}.png'.format(name)
if comment:
graphviz.graph_attributes['graph']['label'] = comment
with PyCallGraph(config=config, output=graphviz):
banana = Banana()
banana.eat()
def run(name, trace_grouper=None, config=None, comment=None):
if not config:
config = Config()
config.trace_filter = GlobbingFilter()
if trace_grouper is not None:
config.trace_grouper = trace_grouper
graphviz = GraphvizOutput()
graphviz.output_file = 'grouper-{}.png'.format(name)
if comment:
graphviz.graph_attributes['graph']['label'] = comment
with PyCallGraph(config=config, output=graphviz):
example_with_submodules.main()
config.trace_filter = GlobbingFilter(exclude=[
'logging.*',
'__main__',
"all",
"any",
"abc.__instancecheck__",
"abc.__subclasscheck__",
"utils.Breakpoints.__init__",
"copy",
"repeat",
"vstack",
"allclose",
"concatenate",
"isclose",
"argmin",
"copyto",
"utils",
"atleast_2d",
"sum",
"nanargmin",
"column_stack",
"result_type",
"threading.RLock",
"cumsum",
"ndim",
"nan_to_num",
"outer",
"typing.new_type",
"apply_along_axis",
"around",
])
import matplotlib
import numpy as np
import pytest
from nottingham_covid_modelling import MODULE_DIR
from nottingham_covid_modelling.optimise_model import run_optimise
from pycallgraph2 import PyCallGraph
from pycallgraph2.output import GraphvizOutput
from pycallgraph2 import Config
from pycallgraph2 import GlobbingFilter
#config.trace_filter = GlobbingFilter(include=['__main__', 'nottingham_covid_modelling.*', 'pints'])
#config.trace_filter = GlobbingFilter(exclude=['argparse', 'argparse.*' 'nottingham_covid_modelling.lib.data', 'nottingham_covid_modelling.lib.data.*', 'unittest', 'unittest.*', 'codecs', 'codecs.*', 'amax', 'argsort', '_bootlocale', 'all', 'logging', 'logging.*', '--new--', '-handle_fromlist', 'warnings', 'warnings.*', 'posixpath', 'posixpath.*', 'os', 'os.*', 'genericpath', 'genericpath.*', 'locle', 'locale.*'])
graphviz = GraphvizOutput()
graphviz.output_file = '/home/uczmh2/optimise_model.png'
testargs = ["run_optimise", "--limit_pints_iterations", "10"]
config = Config(max_depth=5)
config.trace_filter = GlobbingFilter(exclude=[
'pycallgraph.*', 'argparse', 'argparse.*', 'unittest', 'unittest.*',
'logging', 'logging.*', '--new--', 'nottingham_covid_modelling.data.*',
'nottingham_covid_modelling.data.DataLoader',
'nottingham_covid_modelling.data.DataLoader.*'
])
with PyCallGraph(output=graphviz, config=config):
with mock.patch.object(sys, 'argv', testargs):
run_optimise()