changed to gtest
This commit is contained in:
@@ -1,41 +1,41 @@
|
||||
#include <NeuralNetwork/BasisFunction/Linear.h>
|
||||
#include <NeuralNetwork/BasisFunction/Product.h>
|
||||
|
||||
#include <iostream>
|
||||
#include <cassert>
|
||||
#include <chrono>
|
||||
#include "gtest/gtest.h"
|
||||
|
||||
int main() {
|
||||
{
|
||||
NeuralNetwork::BasisFunction::Linear l;
|
||||
assert(39.0==l({1,2,3,5},{1,2,3,5}));
|
||||
}
|
||||
{
|
||||
NeuralNetwork::BasisFunction::Linear l;
|
||||
assert(88.0==l({1,2,3,5,7},{1,2,3,5,7}));
|
||||
}
|
||||
{
|
||||
NeuralNetwork::BasisFunction::Linear l;
|
||||
std::vector<float> w;
|
||||
for(int in=0;in<100;in++) {
|
||||
w.push_back(2);
|
||||
}
|
||||
assert(400.0==l(w,w));
|
||||
}
|
||||
{
|
||||
NeuralNetwork::BasisFunction::Linear l;
|
||||
std::vector<float> w;
|
||||
for(int in=0;in<55;in++) {
|
||||
w.push_back(2);
|
||||
}
|
||||
assert(220.0==l(w,w));
|
||||
}
|
||||
{
|
||||
NeuralNetwork::BasisFunction::Product l;
|
||||
std::vector<float> w({0,0.501,1});
|
||||
std::vector<float> i({0,0.2,0.3});
|
||||
TEST(Linear,FourElements) {
|
||||
NeuralNetwork::BasisFunction::Linear l;
|
||||
ASSERT_EQ(39.0, l({1,2,3,5},{1,2,3,5}));
|
||||
}
|
||||
|
||||
assert(l(w,i) > 0.05999);
|
||||
assert(l(w,i) < 0.06001);
|
||||
TEST(Linear,FiveElements) {
|
||||
NeuralNetwork::BasisFunction::Linear l;
|
||||
ASSERT_EQ(88.0, l({1,2,3,5,7},{1,2,3,5,7}));
|
||||
}
|
||||
|
||||
TEST(Linear,HundredElements) {
|
||||
NeuralNetwork::BasisFunction::Linear l;
|
||||
std::vector<float> w;
|
||||
for(int in=0;in<100;in++) {
|
||||
w.push_back(2);
|
||||
}
|
||||
ASSERT_EQ(400.0, l(w,w));
|
||||
}
|
||||
|
||||
TEST(Linear,FivetyFiveElements) {
|
||||
NeuralNetwork::BasisFunction::Linear l;
|
||||
std::vector<float> w;
|
||||
for(int in=0;in<55;in++) {
|
||||
w.push_back(2);
|
||||
}
|
||||
ASSERT_EQ(220.0, l(w,w));
|
||||
}
|
||||
|
||||
TEST(Product,Product) {
|
||||
NeuralNetwork::BasisFunction::Product p;
|
||||
std::vector<float> w({0,0.501,1});
|
||||
std::vector<float> i({0,0.2,0.3});
|
||||
|
||||
ASSERT_GT(p(w,i), 0.05999);
|
||||
ASSERT_LT(p(w,i), 0.06001);
|
||||
}
|
||||
Reference in New Issue
Block a user