test_cli.sh

#!/bin/bash
set -e

cargo run --release -- help

ZERO_G="target/release/zero_g"

mkdir -p test_data

echo ""
echo "==== Running predict"
$ZERO_G predict \
    -m models/model_28input_256entry_1hash_1bpi.hdf5 \
    -i benches/example_image_7.png

if [ -d "data/MNIST/png" ]; then
    echo ""
    echo "==== Running compute-accuracy"
    $ZERO_G compute-accuracy \
        -m models/model_28input_256entry_1hash_1bpi.hdf5 \
        -t data/MNIST/png/
fi

echo ""
echo "==== Running mock-proof"
$ZERO_G mock-proof \
    -m models/model_28input_256entry_1hash_1bpi.hdf5 \
    -i benches/example_image_7.png \
    -k 14

echo ""
echo "==== Running generate-srs"
$ZERO_G generate-srs \
    -k 14 \
    --srs-path test_data/srs_14

echo ""
echo "==== Running generate-keys"
$ZERO_G generate-keys \
    -m models/model_28input_256entry_1hash_1bpi.hdf5 \
    --srs-path test_data/srs_14 \
    --vk-path test_data/vk \
    --pk-path test_data/pk \
    --circuit-params-path test_data/circuit_params.json

echo ""
echo "==== Running dry-run-evm-verifier"
$ZERO_G dry-run-evm-verifier \
    -m models/model_28input_256entry_1hash_1bpi.hdf5 \
    -i benches/example_image_7.png \
    --srs-path test_data/srs_14 \
    --pk-path test_data/pk

echo ""
echo "==== Running deploy-evm-verifier"
$ZERO_G deploy-evm-verifier \
    --srs-path test_data/srs_14 \
    --vk-path test_data/vk \
    --circuit-params-path test_data/circuit_params.json

echo ""
echo "==== Running proof"
$ZERO_G proof \
    -m models/model_28input_256entry_1hash_1bpi.hdf5 \
    -i benches/example_image_7.png \
    --srs-path test_data/srs_14 \
    --pk-path test_data/pk \
    --proof-path test_data/proof.json

echo ""
echo "==== Running verify"
$ZERO_G verify \
    --srs-path test_data/srs_14 \
    --vk-path test_data/vk \
    --circuit-params-path test_data/circuit_params.json \
    --proof-path test_data/proof.json

echo ""
echo "==== Running submit-proof"
# Using Anvil, the contract address is always the same
$ZERO_G submit-proof \
    --proof-path test_data/proof.json \
    --contract-address 0x5fbdb2315678afecb367f032d93f642f64180aa3