49 lines
1.2 KiB
Go
49 lines
1.2 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"}
|
||
|
|
|
||
|
|
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){
|
||
|
|
|
||
|
|
traitNamesList := []string{"Eye Color", "Lactose Tolerance"}
|
||
|
|
|
||
|
|
for _, traitName := range traitNamesList{
|
||
|
|
|
||
|
|
accuracyInfoBytes, err := geneticPredictionModels.GetPredictionModelTraitAccuracyInfoBytes(traitName)
|
||
|
|
if (err != nil){
|
||
|
|
t.Fatalf("GetGeneticPredictionModelBytes failed: " + err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = geneticPrediction.DecodeBytesToTraitPredictionAccuracyInfoMap(accuracyInfoBytes)
|
||
|
|
if (err != nil){
|
||
|
|
t.Fatalf("DecodeBytesToTraitPredictionAccuracyInfoMap failed: " + err.Error())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|