package geneticPredictionModels_test

import "seekia/resources/geneticPredictionModels"

import "testing"

import "seekia/internal/genetics/geneticPrediction"


func TestGeneticPredictionModels(t *testing.T){

	traitNamesList := []string{"Eye Color", "Lactose Tolerance"}

	for _, traitName := range traitNamesList{

		modelFound, modelBytes := geneticPredictionModels.GetGeneticPredictionModelBytes(traitName)
		if (modelFound == false){
			t.Fatalf("GetGeneticPredictionModelBytes failed to find model for trait: " + traitName)
		}

		_, err := geneticPrediction.DecodeBytesToNeuralNetworkObject(modelBytes)
		if (err != nil){
			t.Fatalf("DecodeBytesToNeuralNetworkObject failed: " + err.Error())
		}
	}
}


func TestGeneticPredictionModelAccuracies(t *testing.T){

	discreteTraitNamesList := []string{"Eye Color", "Lactose Tolerance"}

	for _, traitName := range discreteTraitNamesList{

		accuracyInfoBytes, err := geneticPredictionModels.GetPredictionModelDiscreteTraitAccuracyInfoBytes(traitName)
		if (err != nil){
			t.Fatalf("GetPredictionModelDiscreteTraitAccuracyInfoBytes failed: " + err.Error())
		}

		_, err = geneticPrediction.DecodeBytesToDiscreteTraitPredictionAccuracyInfoMap(accuracyInfoBytes)
		if (err != nil){
			t.Fatalf("DecodeBytesToDiscreteTraitPredictionAccuracyInfoMap failed: " + err.Error())
		}
	}
}