From 74331919d849eba402fe44e1fdef207fab1e7ada Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Juli=C3=A1n=20Gonz=C3=A1lez=20Calder=C3=B3n?= Date: Thu, 19 Sep 2024 14:04:23 -0300 Subject: [PATCH] Fix air_private_input_comparator.py --- vm/src/tests/air_private_input_comparator.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/vm/src/tests/air_private_input_comparator.py b/vm/src/tests/air_private_input_comparator.py index 92ebf987f3..86ced21998 100755 --- a/vm/src/tests/air_private_input_comparator.py +++ b/vm/src/tests/air_private_input_comparator.py @@ -7,6 +7,12 @@ filename1 = sys.argv[1] filename2 = sys.argv[2] +def map_to_vec(batch): + if isinstance(batch, list): + return batch + + return list(batch.values()) + with open(filename1, 'r') as cairo_lang_input_file, open(filename2, 'r') as cairo_vm_input_file: cairo_lang_input = json.load(cairo_lang_input_file) cairo_vm_input = json.load(cairo_vm_input_file) @@ -18,6 +24,16 @@ cairo_vm_input["trace_path"] = "" cairo_vm_input["memory_path"] = "" + # The trace_path & memory_path fields contain a different format for the modulo instance batches + for i, instance in enumerate(cairo_vm_input["add_mod"]["instances"]): + instance["batch"] = map_to_vec(instance["batch"]) + for i, instance in enumerate(cairo_vm_input["mul_mod"]["instances"]): + instance["batch"] = map_to_vec(instance["batch"]) + for i, instance in enumerate(cairo_lang_input["add_mod"]["instances"]): + instance["batch"] = map_to_vec(instance["batch"]) + for i, instance in enumerate(cairo_lang_input["mul_mod"]["instances"]): + instance["batch"] = map_to_vec(instance["batch"]) + if cairo_lang_input == cairo_vm_input: print(f"Comparison succesful for {filename1} vs {filename2}")