#pragma once

#include <stdbool.h>
#include <stddef.h>

#include "psi.h"

void psi_test_section(const char* name);
void psi_test_check(bool ok, const char* name);
bool psi_amps_match(const struct PsiComplex* actual, const struct PsiComplex* expected, size_t n);
void psi_check_circuit(const char* name, struct PsiQuantumCircuit* circuit,
                       const struct PsiComplex* expected, size_t n);
void psi_check_runtimes_agree(const char* name, size_t num_qubits,
                              void (*build)(struct PsiQuantumCircuit*));
int psi_test_summary(void);

void run_clifford_tests(void);
void run_non_clifford_tests(void);
void run_kernel_tests(void);
void run_simd_tests(void);
void run_noise_tests(void);
void run_custom_tests(void);
void run_threading_tests(void);