1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105
|
#include "testDatasetManager.h"
#include <public.h>
#include <datasetManager.h>
#include <mymaths.h>
using namespace std;
void TestDatasetManager::initDatasetManager()
{
DatasetManager manager;
QCOMPARE(manager.GetCount(), 0);
}
void TestDatasetManager::addData_data()
{
QTest::addColumn<int>("count");
QTest::addColumn<int>("dim");
QTest::newRow("1x 1D") << 1 << 1;
QTest::newRow("10000x 1D") << 10000 << 1;
QTest::newRow("1x 10D") << 1 << 10;
QTest::newRow("1x 10000D") << 1 << 10000;
QTest::newRow("10x 10D") << 10 << 10;
QTest::newRow("1000x 1000D") << 1000 << 1000;
// QTest::newRow("100000x 2D") << 100000 << 2;
}
void TestDatasetManager::addData()
{
QFETCH(int, count);
QFETCH(int, dim);
DatasetManager manager;
fvec sample(dim);
FOR(i, count) {
FOR(d, dim) sample[d] = drand48()*2 - 1.f;
manager.AddSample(sample);
}
QCOMPARE(manager.GetCount(), count);
QCOMPARE(manager.GetDimCount(), dim);
}
void TestDatasetManager::addBulkData_data()
{
QTest::addColumn<int>("count");
QTest::addColumn<int>("dim");
QTest::newRow("1x 1D") << 1 << 1;
QTest::newRow("10000x 1D") << 10000 << 1;
QTest::newRow("1x 10D") << 1 << 10;
QTest::newRow("1x 10000D") << 1 << 10000;
QTest::newRow("10x 10D") << 10 << 10;
QTest::newRow("1000x 1000D") << 1000 << 1000;
QTest::newRow("100000x 1000D") << 100000 << 1000;
}
void TestDatasetManager::addBulkData()
{
QFETCH(int, count);
QFETCH(int, dim);
DatasetManager manager;
fvec sample(dim);
vector<fvec> samples;
FOR(i, count) {
FOR(d, dim) sample[d] = drand48()*2 - 1.f;
samples.push_back(sample);
}
manager.AddSamples(samples);
QCOMPARE(manager.GetCount(), count);
QCOMPARE(manager.GetDimCount(), dim);
}
void TestDatasetManager::addDataset_data()
{
QTest::addColumn<int>("count");
QTest::addColumn<int>("dim");
QTest::newRow("1x 1D") << 1 << 1;
QTest::newRow("10000x 1D") << 10000 << 1;
QTest::newRow("1x 10D") << 1 << 10;
QTest::newRow("1x 10000D") << 1 << 10000;
QTest::newRow("10x 10D") << 10 << 10;
QTest::newRow("1000x 1000D") << 1000 << 1000;
QTest::newRow("100000x 1000D") << 100000 << 1000;
}
void TestDatasetManager::addDataset()
{
QFETCH(int, count);
QFETCH(int, dim);
DatasetManager manager1, manager2;
fvec sample(dim);
vector<fvec> samples;
FOR(i, count) {
FOR(d, dim) sample[d] = drand48()*2 - 1.f;
samples.push_back(sample);
}
manager1.AddSamples(samples);
manager2.AddSamples(manager1);
manager2.AddSamples(manager1);
QCOMPARE(manager2.GetCount(), 2*count);
QCOMPARE(manager2.GetDimCount(), dim);
}
QTEST_MAIN(TestDatasetManager)
|