28 lines
803 B
Go
28 lines
803 B
Go
|
package geneticPrediction_test
|
||
|
|
|
||
|
|
|
||
|
|
import "seekia/internal/genetics/geneticPrediction"
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
|
||
|
|
// We test the encoding/decoding of a neural network object
|
||
|
|
func TestEncodeNeuralNetwork(t *testing.T){
|
||
|
|
|
||
|
|
neuralNetworkObject, err := geneticPrediction.GetNewUntrainedNeuralNetworkObject("Eye Color")
|
||
|
|
if (err != nil) {
|
||
|
|
t.Fatalf("GetNewUntrainedNeuralNetworkObject failed: " + err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
neuralNetworkBytes, err := geneticPrediction.EncodeNeuralNetworkObjectToBytes(*neuralNetworkObject)
|
||
|
|
if (err != nil){
|
||
|
|
t.Fatalf("EncodeNeuralNetworkObjectToBytes failed: " + err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = geneticPrediction.DecodeBytesToNeuralNetworkObject(neuralNetworkBytes)
|
||
|
|
if (err != nil){
|
||
|
|
t.Fatalf("DecodeBytesToNeuralNetworkObject failed: " + err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
//TODO: Verify values are the same
|
||
|
|
}
|