def test_greedy(cascades_on_grid):
for g, infection_times, source, obs_nodes, true_tree, model, q, i in cascades_on_grid:
print(model, q, i)
root = earliest_obs_node(obs_nodes, infection_times)
tree = find_tree_greedy(g,
root,
infection_times,
source,
obs_nodes,
debug=False,
verbose=True)
assert is_feasible(tree, root, obs_nodes, infection_times)
def test_temporal_bfs(cascades_on_grid):
for g, infection_times, source, obs_nodes, true_tree, model, q, i in cascades_on_grid:
print(model, q, i)
root = earliest_obs_node(obs_nodes, infection_times)
tree = temporal_bfs(
g,
root,
infection_times,
source,
obs_nodes,
debug=False,
verbose=False,
)
assert is_feasible(tree, root, obs_nodes, infection_times)
def test_closure(cascades_on_grid):
for g, infection_times, source, obs_nodes, true_tree, model, q, i in cascades_on_grid:
print(model, q, i)
root = earliest_obs_node(obs_nodes, infection_times)
tree = find_tree_by_closure(
g,
root,
infection_times,
obs_nodes,
strictly_smaller=False,
debug=False,
verbose=False,
)
assert is_feasible(tree, root, obs_nodes, infection_times)
def test_mst(cascades_on_tree):
for g, infection_times, source, obs_nodes, true_tree, model, q, i in cascades_on_tree:
print(model, q, i)
root = earliest_obs_node(obs_nodes, infection_times)
tree = steiner_tree_mst(
g,
root,
infection_times,
source,
obs_nodes,
closure_builder=build_closure,
strictly_smaller=False,
debug=False,
verbose=False,
)
assert is_feasible(tree, root, obs_nodes, infection_times)