63 lines
1.8 KiB
Go
63 lines
1.8 KiB
Go
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", "Height"}
|
|
|
|
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())
|
|
}
|
|
}
|
|
|
|
numericTraitNamesList := []string{"Height"}
|
|
|
|
for _, traitName := range numericTraitNamesList{
|
|
|
|
accuracyInfoBytes, err := geneticPredictionModels.GetPredictionModelNumericTraitAccuracyInfoBytes(traitName)
|
|
if (err != nil){
|
|
t.Fatalf("GetPredictionModelNumericTraitAccuracyInfoBytes failed: " + err.Error())
|
|
}
|
|
|
|
_, err = geneticPrediction.DecodeBytesToNumericTraitPredictionAccuracyInfoMap(accuracyInfoBytes)
|
|
if (err != nil){
|
|
t.Fatalf("DecodeBytesToNumericTraitPredictionAccuracyInfoMap failed: " + err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
|