#include <stdlib.h>
#include <stdio.h>
#include <stdint.h>
#include <inttypes.h>

#include "../src/include/colors.h"
#include "../src/cnn/include/neuron_io.h"
#include "../src/cnn/include/creation.h"
#include "../src/cnn/include/utils.h"
#include "../src/include/colors.h"


int main() {
    printf("Création du réseau\n");
    Network* network = create_network_lenet5(0, 0, 3, 2, 32, 1);
    printf(GREEN "OK\n" RESET);

    printf("Écriture du réseau\n");
    write_network((char*)".test-cache/cnn_neuron_io.bin", network);
    printf(GREEN "OK\n" RESET);

    printf("Vérification de l'accès en lecture\n");
    Network* network2 = read_network((char*)".test-cache/cnn_neuron_io.bin");
    printf(GREEN "OK\n" RESET);

    printf("Vérification de l'égalité des réseaux\n");
    if (! equals_networks(network, network2)) {
        printf_error(RED "Les deux réseaux obtenus ne sont pas égaux.\n" RESET);
        exit(1);
    }
    printf(GREEN "OK\n" RESET);

    return 0;
}