Add generated qed compton graph to execution test

2023-12-07 00:49:30 +01:00
parent 86799644c4
commit 5e5e29dc69
1 changed files with 6 additions and 2 deletions
--- a/test/unit_tests_qedmodel.jl
+++ b/test/unit_tests_qedmodel.jl
@@ -279,9 +279,13 @@ end

    insert_edge!(graph, sum_node, d_exit, track = false)

-    compton_function = get_compute_function(graph, process, machine)
-
    input = [gen_process_input(process) for _ in 1:1000]

+    compton_function = get_compute_function(graph, process, machine)
+    @test isapprox(compton_function.(input), compton_groundtruth.(input))
+
+    graph_generated = gen_graph(process)
+
+    compton_function = get_compute_function(graph_generated, process, machine)
    @test isapprox(compton_function.(input), compton_groundtruth.(input))
 end