// Table manipulations
TEST(TStrIntH, ManipulateTable) {
const int64 NElems = 1000000;
int DDist = 10;
const char *FName = "test.hashstr.dat";
TStrIntH TableStr;
TStrIntH TableStr1;
TStrIntH TableStr2;
int i;
int d;
int n;
int Id;
int Key;
TStr KeyStr;
int64 KeySumVal;
int64 DatSumVal;
int64 KeySum;
int64 DatSum;
int64 KeySumDel;
int64 DatSumDel;
int DelCount;
int Count;
char s[32];
// add table elements
d = Prime(NElems);
n = d;
KeySumVal = 0;
DatSumVal = 0;
for (i = 0; i < NElems; i++) {
sprintf(s,"%d",n);
TStr Str = TStr(s);
TableStr.AddDat(Str,n+1);
KeySumVal += n;
DatSumVal += (n+1);
//printf("add %d %d\n", n, n+1);
n = (n + d) % NElems;
}
EXPECT_EQ(0,TableStr.Empty());
EXPECT_EQ(NElems,TableStr.Len());
EXPECT_EQ(0,(NElems-1)*(NElems)/2 - KeySumVal);
EXPECT_EQ(0,(NElems)*(NElems+1)/2 - DatSumVal);
// verify elements by successive keys
KeySum = 0;
DatSum = 0;
for (i = 0; i < NElems; i++) {
sprintf(s,"%d",i);
TStr Str = TStr(s);
Id = TableStr.GetKeyId(s);
EXPECT_EQ(1,Id >= 0);
KeyStr = TableStr.GetKey(Id);
Key = atoi(KeyStr.CStr());
//printf("vrfy %d %s %d %s %d %d\n", i, Str.CStr(), Id, KeyStr.CStr(), Key, (int) TableStr.GetDat(KeyStr));
EXPECT_EQ(0,TableStr.GetDat(KeyStr)-Key-1);
KeySum += Key;
DatSum += TableStr.GetDat(KeyStr);
}
EXPECT_EQ(0,KeySumVal - KeySum);
EXPECT_EQ(0,DatSumVal - DatSum);
// verify elements by distant keys
KeySum = 0;
DatSum = 0;
n = Prime(d);
for (i = 0; i < NElems; i++) {
sprintf(s,"%d",i);
TStr Str = TStr(s);
Id = TableStr.GetKeyId(s);
EXPECT_EQ(1,Id >= 0);
KeyStr = TableStr.GetKey(Id);
Key = atoi(KeyStr.CStr());
EXPECT_EQ(0,TableStr.GetDat(KeyStr)-Key-1);
KeySum += Key;
DatSum += TableStr.GetDat(KeyStr);
n = (n + d) % NElems;
}
EXPECT_EQ(0,KeySumVal - KeySum);
EXPECT_EQ(0,DatSumVal - DatSum);
// verify elements by iterator
KeySum = 0;
DatSum = 0;
for (TStrIntH::TIter It = TableStr.BegI(); It < TableStr.EndI(); It++) {
KeyStr = It.GetKey();
Key = atoi(KeyStr.CStr());
EXPECT_EQ(0,It.GetDat()-Key-1);
KeySum += Key;
DatSum += It.GetDat();
}
EXPECT_EQ(0,KeySumVal - KeySum);
EXPECT_EQ(0,DatSumVal - DatSum);
// verify elements by key index
KeySum = 0;
DatSum = 0;
Id = TableStr.FFirstKeyId();
while (TableStr.FNextKeyId(Id)) {
EXPECT_EQ(1,Id >= 0);
KeyStr = TableStr.GetKey(Id);
Key = atoi(KeyStr.CStr());
EXPECT_EQ(0,TableStr.GetDat(KeyStr)-Key-1);
KeySum += Key;
DatSum += TableStr.GetDat(KeyStr);
}
EXPECT_EQ(0,KeySumVal - KeySum);
EXPECT_EQ(0,DatSumVal - DatSum);
// delete elements
DelCount = 0;
KeySumDel = 0;
DatSumDel = 0;
for (n = 0; n < NElems; n += DDist) {
sprintf(s,"%d",n);
TStr Str = TStr(s);
Id = TableStr.GetKeyId(Str);
//printf("del %d %d %d\n", n, Id, (int) TableStr[Id]);
KeySumDel += n;
DatSumDel += TableStr[Id];
TableStr.DelKeyId(Id);
DelCount++;
}
EXPECT_EQ(0,TableStr.Empty());
EXPECT_EQ(NElems-DelCount,TableStr.Len());
// verify elements by iterator
KeySum = 0;
DatSum = 0;
Count = 0;
for (TStrIntH::TIter It = TableStr.BegI(); It < TableStr.EndI(); It++) {
KeyStr = It.GetKey();
Key = atoi(KeyStr.CStr());
EXPECT_EQ(0,It.GetDat()-Key-1);
//printf("get %d %d\n", (int) It.GetKey(), (int) It.GetDat());
KeySum += Key;
DatSum += It.GetDat();
Count++;
}
EXPECT_EQ(NElems-DelCount,Count);
EXPECT_EQ(0,KeySumVal - KeySumDel - KeySum);
EXPECT_EQ(0,DatSumVal - DatSumDel - DatSum);
// assignment
TableStr1 = TableStr;
EXPECT_EQ(0,TableStr1.Empty());
EXPECT_EQ(NElems-DelCount,TableStr1.Len());
// verify elements by iterator
KeySum = 0;
DatSum = 0;
Count = 0;
for (TStrIntH::TIter It = TableStr1.BegI(); It < TableStr1.EndI(); It++) {
KeyStr = It.GetKey();
Key = atoi(KeyStr.CStr());
EXPECT_EQ(0,It.GetDat()-Key-1);
//printf("get %d %d\n", (int) It.GetKey(), (int) It.GetDat());
KeySum += Key;
DatSum += It.GetDat();
Count++;
}
EXPECT_EQ(NElems-DelCount,Count);
EXPECT_EQ(0,KeySumVal - KeySumDel - KeySum);
EXPECT_EQ(0,DatSumVal - DatSumDel - DatSum);
// saving and loading
{
TFOut FOut(FName);
TableStr.Save(FOut);
FOut.Flush();
}
{
TFIn FIn(FName);
TableStr2.Load(FIn);
}
EXPECT_EQ(NElems-DelCount,TableStr2.Len());
// verify elements by iterator
KeySum = 0;
DatSum = 0;
Count = 0;
for (TStrIntH::TIter It = TableStr2.BegI(); It < TableStr2.EndI(); It++) {
KeyStr = It.GetKey();
Key = atoi(KeyStr.CStr());
EXPECT_EQ(0,It.GetDat()-Key-1);
//printf("get %d %d\n", (int) It.GetKey(), (int) It.GetDat());
KeySum += Key;
DatSum += It.GetDat();
Count++;
}
EXPECT_EQ(NElems-DelCount,Count);
EXPECT_EQ(0,KeySumVal - KeySumDel - KeySum);
EXPECT_EQ(0,DatSumVal - DatSumDel - DatSum);
// remove all elements
for (i = 0; i < Count; i++) {
Id = TableStr.GetRndKeyId(TInt::Rnd, 0.5);
TableStr.DelKeyId(Id);
}
EXPECT_EQ(0,TableStr.Len());
EXPECT_EQ(1,TableStr.Empty());
// verify elements by iterator
KeySum = 0;
DatSum = 0;
Count = 0;
for (TStrIntH::TIter It = TableStr.BegI(); It < TableStr.EndI(); It++) {
KeyStr = It.GetKey();
Key = atoi(KeyStr.CStr());
EXPECT_EQ(0,It.GetDat()-Key-1);
//printf("get %d %d\n", (int) It.GetKey(), (int) It.GetDat());
KeySum += Key;
DatSum += It.GetDat();
Count++;
}
EXPECT_EQ(0,Count);
EXPECT_EQ(0,KeySum);
EXPECT_EQ(0,DatSum);
// clear the table
TableStr1.Clr();
EXPECT_EQ(0,TableStr1.Len());
EXPECT_EQ(1,TableStr1.Empty());
// verify elements by iterator
KeySum = 0;
DatSum = 0;
Count = 0;
for (TStrIntH::TIter It = TableStr1.BegI(); It < TableStr1.EndI(); It++) {
KeyStr = It.GetKey();
Key = atoi(KeyStr.CStr());
EXPECT_EQ(0,It.GetDat()-Key-1);
//printf("get %d %d\n", (int) It.GetKey(), (int) It.GetDat());
KeySum += Key;
DatSum += It.GetDat();
Count++;
}
EXPECT_EQ(0,Count);
EXPECT_EQ(0,KeySum);
EXPECT_EQ(0,DatSum);
}
