GroupByTest.cpp

#include <vector>
#include <string>

#include <gtest/gtest.h>
#include "CommonTests.h"

#include "boolinq.h"

using namespace boolinq;

//////////////////////////////////////////////////////////////////////////

TEST(GroupBy, Ints)
{
    int arr[] = {0,1,2,3,4,5,6,7,8,9};

    int ans_0[] = {0,3,6,9};
    int ans_1[] = {1,4,7};
    int ans_2[] = {2,5,8};

    auto dst = from(arr).groupBy([](int a){return a % 3;});

    EXPECT_EQ(0, dst.elementAt(0).first);
    EXPECT_EQ(1, dst.elementAt(1).first);
    EXPECT_EQ(2, dst.elementAt(2).first);
    CheckRangeEqArray(dst.elementAt(0).second, ans_0);
    CheckRangeEqArray(dst.elementAt(1).second, ans_1);
    CheckRangeEqArray(dst.elementAt(2).second, ans_2);

    EXPECT_THROW(dst.elementAt(3), LinqEndException);
}

//////////////////////////////////////////////////////////////////////////

TEST(GroupBy, CountChildrenByAge)
{
    struct Child
    {
        std::string name;
        int age;

        bool operator == (const Child & rhs) const
        {
            return (name == rhs.name) && (age == rhs.age);
        }
    };  

    Child children[] = {
        {"Piter", 12},
        {"Bella", 14},
        {"Torry", 15},
        {"Holly", 12},
        {"Zamza", 13},
    };

    Child ans_false[] = {
        {"Bella", 14},
        {"Torry", 15},
    };

    Child ans_true[] = {
        {"Piter", 12},
        {"Holly", 12},
        {"Zamza", 13},
    };

    auto dst = from(children).groupBy([](const Child & a){return a.age < 14;});

    EXPECT_EQ(true, dst.elementAt(0).first);
    EXPECT_EQ(false, dst.elementAt(1).first);
    CheckRangeEqArray(dst.elementAt(0).second, ans_true);
    CheckRangeEqArray(dst.elementAt(1).second, ans_false);

    EXPECT_THROW(dst.elementAt(2), LinqEndException);
}

//////////////////////////////////////////////////////////////////////////

TEST(GroupBy, CountChildrenByName)
{
    struct Child
    {
        std::string name;
        int age;

        bool operator == (const Child & rhs) const
        {
            return (name == rhs.name) && (age == rhs.age);
        }
    };

    Child children[] = {
        {"Tom", 12},
        {"Bella", 14},
        {"Torry", 15},
        {"Tom", 11},
        {"Bella", 13},
    };

    Child ans_1[] = {
        {"Tom", 12},
        {"Tom", 11},
    };

    Child ans_2[] = {
        {"Bella", 14},
        {"Bella", 13},
    };

    Child ans_3[] = {
        {"Torry", 15},
    };

    auto dst = from(children).groupBy([](const Child & a){return a.name;});

    EXPECT_EQ("Tom", dst.elementAt(0).first);
    EXPECT_EQ("Bella", dst.elementAt(1).first);
    EXPECT_EQ("Torry", dst.elementAt(2).first);
//    CheckRangeEqArray(dst.elementAt(0).second, ans_1);
//    CheckRangeEqArray(dst.elementAt(1).second, ans_2);
//    CheckRangeEqArray(dst.elementAt(2).second, ans_3);

    EXPECT_THROW(dst.elementAt(3), LinqEndException);
}