Adds tests/hydra/starknet/test_calldata.py

tests/hydra/starknet/test_calldata.py

@@ -0,0 +1,62 @@
+import random
+
+import pytest
+
+from garaga.definitions import CURVES, STARK, CurveID, G1Point
+from garaga.starknet.tests_and_calldata_generators.msm import MSMCalldataBuilder
+
+# Define the curves to be tested
+curves = list(CurveID)
+
+
+@pytest.mark.parametrize("curve_id", curves)
+@pytest.mark.parametrize("msm_size", range(1, 2))
+@pytest.mark.parametrize("include_digits_decomposition", [True, False])
+@pytest.mark.parametrize("include_points_and_scalars", [True, False])
+@pytest.mark.parametrize("serialize_as_pure_felt252_array", [True, False])
+@pytest.mark.parametrize("risc0_mode", [True, False])
+def test_msm_calldata_builder(
+    curve_id,
+    msm_size,
+    include_digits_decomposition,
+    include_points_and_scalars,
+    serialize_as_pure_felt252_array,
+    risc0_mode,
+):
+    curve = CURVES[curve_id.value]
+    order = curve.n
+
+    scalar_limit = min(order, 2**128) if risc0_mode else order
+
+    points = [G1Point.gen_random_point(curve_id) for _ in range(msm_size)]
+    scalars = [random.randint(0, scalar_limit - 1) for _ in range(msm_size)]
+
+    msm = MSMCalldataBuilder(
+        points=points,
+        scalars=scalars,
+        curve_id=curve_id,
+    )
+
+    calldata1 = msm.serialize_to_calldata(
+        include_digits_decomposition=include_digits_decomposition,
+        include_points_and_scalars=include_points_and_scalars,
+        serialize_as_pure_felt252_array=serialize_as_pure_felt252_array,
+        risc0_mode=risc0_mode,
+        use_rust=False,
+    )
+
+    calldata2 = msm.serialize_to_calldata(
+        include_digits_decomposition=include_digits_decomposition,
+        include_points_and_scalars=include_points_and_scalars,
+        serialize_as_pure_felt252_array=serialize_as_pure_felt252_array,
+        risc0_mode=risc0_mode,
+        use_rust=True,
+    )
+
+    calldata2 = [value % STARK for value in calldata2]
+
+    assert calldata1 == calldata2
+
+
+if __name__ == "__main__":
+    pytest.main()