def METHOD_NAME(self):
    model = self._get_model()
    graph = secir.ModelGraph()
    graph.add_node(0, model)
    graph.add_node(1, model)
    graph.add_edge(0, 1, 0.01 * np.ones(10))
    graph.add_edge(1, 0, 0.01 * np.ones(10))
    study = secir.ParameterStudy(graph, t0=1, tmax=10, dt=0.5, num_runs=3)
    self.assertEqual(study.secir_model_graph.num_nodes, 2)
    self.assertEqual(study.secir_model_graph.num_edges, 2)