Files
mlpack/fastlib/u/nvasil/dataset/binary_dataset_unit.cc
T

102 lines
2.5 KiB
C++

/*
* =====================================================================================
*
* Filename: binary_dataset.cc
*
* Description:
*
* Version: 1.0
* Created: 04/24/2007 07:36:16 PM EDT
* Revision: none
* Compiler: gcc
*
* Author: Nikolaos Vasiloglou (NV), nvasil@ieee.org
* Company: Georgia Tech Fastlab-ESP Lab
*
* =====================================================================================
*/
#include <string>
#include <unistd.h>
#include "fastlib/fastlib.h"
#include "u/nvasil/test/test.h"
#include "u/nvasil/dataset/binary_dataset.h"
using namespace std;
template<typename PRECISION>
class BinaryDatasetTest {
public:
typedef PRECISION Precision_t;
void Init() {
data_file_="data";
dimension_=10;
num_of_points_=8000;
dataset_.Init(data_file_,
num_of_points_,
dimension_);
for(uint64 i=0; i<num_of_points_; i++) {
for(int32 j=0; j<dimension_; j++) {
dataset_.At(i,j)=1.0*16.3;
}
dataset_.set_id(i,i);
}
}
void Destruct() {
dataset_.Destruct();
unlink(data_file_.c_str());
unlink(data_file_.append(".ind").c_str());
data_file_="";
}
void FillDataset() {
for(uint64 i=0; i<num_of_points_; i++) {
for(int32 j=0; j<dimension_; j++) {
TEST_ASSERT(dataset_.At(i,j)==1.0*16.3) ;
}
}
dataset_.Destruct();
dataset_.Init(data_file_);
for(uint64 i=0; i<num_of_points_; i++) {
for(int32 j=0; j<dimension_; j++) {
TEST_ASSERT(dataset_.At(i,j)==1.0*16.3) ;
}
}
}
void Swap() {
index_t i=140;
index_t j=1456;
Precision_t point_i[dimension_];
memcpy(point_i, dataset_.At(i), dimension_*sizeof(Precision_t));
uint64 id_i=dataset_.get_id(i);
Precision_t point_j[dimension_];
uint64 id_j=dataset_.get_id(j);
memcpy(point_j, dataset_.At(j), dimension_*sizeof(Precision_t));
dataset_.Swap(i,j);
TEST_ASSERT(memcmp(dataset_.At(i), point_j,
dimension_*sizeof(Precision_t))==0);
TEST_ASSERT(memcmp(dataset_.At(j), point_i,
dimension_*sizeof(Precision_t))==0);
TEST_ASSERT(dataset_.get_id(i)==id_j);
TEST_ASSERT(dataset_.get_id(j)==id_i);
}
void TestAll() {
Init();
FillDataset();
Destruct();
Init();
Swap();
Destruct();
}
private:
string data_file_;
uint64 num_of_points_;
int32 dimension_;
BinaryDataset<Precision_t> dataset_;
};
int main(int argc, char *argv[]) {
BinaryDatasetTest<double> binary_dataset_tester;
binary_dataset_tester.TestAll();
}