#include "Data/ArrayData.h"
#include <QObject>
#include <QtTest>

using DataContainer = QVector<QVector<double> >;

class TestTwoDimArrayData : public QObject {
    Q_OBJECT
private slots:
    /// Tests @sa ArrayData::data(int componentIndex)
    void testDataByComponentIndex_data();
    void testDataByComponentIndex();

    /// Tests @sa ArrayData::add()
    void testAdd_data();
    void testAdd();

};

void TestTwoDimArrayData::testDataByComponentIndex_data()
{
    // Test structure
    QTest::addColumn<DataContainer>("inputData");       // array data's input
    QTest::addColumn<int>("componentIndex");            // component index to test
    QTest::addColumn<QVector<double> >("expectedData"); // expected data

    // Test cases
    auto inputData
        = DataContainer{{1., 2., 3., 4., 5.}, {6., 7., 8., 9., 10.}, {11., 12., 13., 14., 15.}};

    QTest::newRow("validIndex1") << inputData << 0 << QVector<double>{1., 2., 3., 4., 5.};
    QTest::newRow("validIndex2") << inputData << 1 << QVector<double>{6., 7., 8., 9., 10.};
    QTest::newRow("validIndex3") << inputData << 2 << QVector<double>{11., 12., 13., 14., 15.};
    QTest::newRow("invalidIndex1") << inputData << -1 << QVector<double>{};
    QTest::newRow("invalidIndex2") << inputData << 3 << QVector<double>{};
}

void TestTwoDimArrayData::testDataByComponentIndex()
{
    QFETCH(DataContainer, inputData);
    QFETCH(int, componentIndex);
    QFETCH(QVector<double>, expectedData);

    ArrayData<2> arrayData{inputData};
    QVERIFY(arrayData.data(componentIndex) == expectedData);
}

void TestTwoDimArrayData::testAdd_data()
{
    // Test structure
    QTest::addColumn<DataContainer>("inputData");    // array's data input
    QTest::addColumn<DataContainer>("otherData");    // array data's input to merge with
    QTest::addColumn<bool>("prepend");               // prepend or append merge
    QTest::addColumn<DataContainer>("expectedData"); // expected data after merge

    // Test cases
    auto inputData
        = DataContainer{{1., 2., 3., 4., 5.}, {11., 12., 13., 14., 15.}, {21., 22., 23., 24., 25.}};

    auto vectorContainer = DataContainer{{6., 7., 8.}, {16., 17., 18.}, {26., 27., 28}};
    auto tensorContainer = DataContainer{{6., 7., 8.},    {16., 17., 18.}, {26., 27., 28},
                                         {36., 37., 38.}, {46., 47., 48.}, {56., 57., 58}};

    QTest::newRow("appendMerge") << inputData << vectorContainer << false
                                 << DataContainer{{1., 2., 3., 4., 5., 6., 7., 8.},
                                                  {11., 12., 13., 14., 15., 16., 17., 18.},
                                                  {21., 22., 23., 24., 25., 26., 27., 28}};
    QTest::newRow("prependMerge") << inputData << vectorContainer << true
                                  << DataContainer{{6., 7., 8., 1., 2., 3., 4., 5.},
                                                   {16., 17., 18., 11., 12., 13., 14., 15.},
                                                   {26., 27., 28, 21., 22., 23., 24., 25.}};
    QTest::newRow("invalidMerge") << inputData << tensorContainer << false << inputData;
}

void TestTwoDimArrayData::testAdd()
{
    QFETCH(DataContainer, inputData);
    QFETCH(DataContainer, otherData);
    QFETCH(bool, prepend);
    QFETCH(DataContainer, expectedData);

    ArrayData<2> arrayData{inputData};
    ArrayData<2> other{otherData};

    arrayData.add(other, prepend);

    for (auto i = 0; i < expectedData.size(); ++i) {
        QVERIFY(arrayData.data(i) == expectedData.at(i));
    }
}

QTEST_MAIN(TestTwoDimArrayData)
#include "TestTwoDimArrayData.moc"