extern crate assert;
extern crate dwt;

use dwt::{Operation, transform};

#[test]
fn forward_haar_40() {
    let data = vec![
        4.018080337519417e-01, 7.596669169084191e-02, 2.399161535536580e-01, 1.233189348351655e-01,
        1.839077882824167e-01, 2.399525256649028e-01, 4.172670690843695e-01, 4.965443032574213e-02,
        9.027161099152811e-01, 9.447871897216460e-01, 4.908640924680799e-01, 4.892526384000189e-01,
        3.377194098213772e-01, 9.000538464176620e-01, 3.692467811202150e-01, 1.112027552937874e-01,
        7.802520683211379e-01, 3.897388369612534e-01, 2.416912859138327e-01, 4.039121455881147e-01,
        9.645452516838859e-02, 1.319732926063351e-01, 9.420505907754851e-01, 9.561345402298023e-01,
        5.752085950784656e-01, 5.977954294715582e-02, 2.347799133724063e-01, 3.531585712220711e-01,
        8.211940401979591e-01, 1.540343765155505e-02, 4.302380165780784e-02, 1.689900294627044e-01,
        6.491154749564521e-01, 7.317223856586703e-01, 6.477459631363067e-01, 4.509237064309449e-01,
        5.470088922863450e-01, 2.963208056077732e-01, 7.446928070741562e-01, 1.889550150325445e-01,
    ];

    let expected_data_3 = [
         6.122808015937273e-01,  1.607198143231635e+00,
         1.393780752232782e+00,  8.031099375749208e-01,
         1.504894721501849e+00, -1.759705918896182e-02,
         3.922311549573817e-01, -1.099616848086838e-01,
         6.162977017331306e-02,  2.483818670932161e-01,
         5.726981852698007e-02, -2.153059273139610e-02,
         4.336932843844141e-01,  3.786618599125184e-01,
         2.621937368902220e-01, -8.348786566152820e-01,
         2.352482671557193e-02,  3.122918233645010e-01,
         1.410840955239354e-01, -4.515906210629123e-02,
         2.304046225623291e-01,  8.244668402333712e-02,
        -3.962961385297514e-02,  2.599413897161061e-01,
        -2.974874582292109e-02,  1.139470099096540e-03,
        -3.976304934119496e-01,  1.824646805065435e-01,
         2.761345540376455e-01, -1.147074699255962e-01,
        -2.511556131475992e-02, -9.958856165036223e-03,
         3.644633779826036e-01, -8.370635171326010e-02,
         5.697799992769564e-01, -8.907157388133179e-02,
        -5.841190673041013e-02,  1.391743524048007e-01,
         1.772632460530991e-01,  3.929659613142630e-01,
    ];

    {
        let mut data = data.clone();
        transform(&mut data, Operation::Forward, &dwt::wavelet::Haar::new(), 3);
        assert::close(&data, &expected_data_3[..], 1e-14);
    }
}

#[test]
fn forward_haar_64() {
    let data = vec![
        4.217612826262750e-01, 9.157355251890671e-01, 7.922073295595544e-01, 9.594924263929030e-01,
        6.557406991565868e-01, 3.571167857418955e-02, 8.491293058687771e-01, 9.339932477575505e-01,
        6.787351548577735e-01, 7.577401305783334e-01, 7.431324681249162e-01, 3.922270195341682e-01,
        6.554778901775566e-01, 1.711866878115618e-01, 7.060460880196088e-01, 3.183284637742068e-02,
        2.769229849608900e-01, 4.617139063115394e-02, 9.713178123584754e-02, 8.234578283272926e-01,
        6.948286229758170e-01, 3.170994800608605e-01, 9.502220488383549e-01, 3.444608050290876e-02,
        4.387443596563982e-01, 3.815584570930084e-01, 7.655167881490024e-01, 7.951999011370632e-01,
        1.868726045543786e-01, 4.897643957882311e-01, 4.455862007108995e-01, 6.463130101112646e-01,
        7.093648308580726e-01, 7.546866819823609e-01, 2.760250769985784e-01, 6.797026768536748e-01,
        6.550980039738407e-01, 1.626117351946306e-01, 1.189976815583766e-01, 4.983640519821430e-01,
        9.597439585160811e-01, 3.403857266661332e-01, 5.852677509797773e-01, 2.238119394911370e-01,
        7.512670593056529e-01, 2.550951154592691e-01, 5.059570516651424e-01, 6.990767226566860e-01,
        8.909032525357985e-01, 9.592914252054443e-01, 5.472155299638031e-01, 1.386244428286791e-01,
        1.492940055590575e-01, 2.575082541237365e-01, 8.407172559836625e-01, 2.542821789715310e-01,
        8.142848260688164e-01, 2.435249687249893e-01, 9.292636231872278e-01, 3.499837659848087e-01,
        1.965952504312082e-01, 2.510838579760311e-01, 6.160446761466392e-01, 4.732888489027293e-01,
    ];

    let expected_data_6 = [
         4.147168492759418e+00,  1.253279360754873e-01,
         4.084096467714702e-01,  4.658948064130208e-02,
         3.568483024108910e-01, -2.273188749167802e-01,
        -1.164386463345503e-01,  4.094163193731559e-02,
         2.173015620707741e-01,  3.561312405386555e-01,
        -2.661946778248070e-01,  2.165455455832754e-01,
         3.481467793065587e-01, -3.612840951056551e-02,
         3.656565682202418e-01,  2.828584634582116e-01,
        -2.071014740685578e-01, -5.458350879477758e-01,
         1.505578988885113e-01,  4.439282179604442e-02,
        -2.987476169855482e-01,  1.362998684770705e-02,
        -3.702069362683296e-01, -2.076311052397773e-01,
         2.541618794940903e-01,  1.001740028139758e-01,
         2.455249973556499e-01, -9.933579977845319e-02,
         5.821773524743806e-01, -3.440985876361998e-01,
        -1.107187971891155e-01, -3.208272083210646e-01,
        -3.492925366476388e-01, -1.182884263623091e-01,
         4.384267249862666e-01, -6.000786878777276e-02,
        -5.586495407948650e-02,  2.481276222538253e-01,
         3.424455932619815e-01,  4.767407551309557e-01,
         1.631660171201637e-01, -5.135900732507805e-01,
         2.670948384069484e-01,  6.475513972576711e-01,
         4.043653949084614e-02, -2.098913048058437e-02,
        -2.141768395471971e-01, -1.419352880929379e-01,
        -3.204738826591147e-02, -2.854431682706484e-01,
         3.482403802950402e-01, -2.682525330807729e-01,
         4.379524057248081e-01,  2.555878554029040e-01,
         3.508465461282888e-01, -1.365562289386354e-01,
        -4.835774064766307e-02,  2.889175284456297e-01,
        -7.651902898109114e-02,  4.146722196809334e-01,
         4.035881655568866e-01,  4.096127152326055e-01,
        -3.852926389235672e-02,  1.009436134980640e-01,
    ];

    let expected_data_2 = [
         1.544598281883899e+00,  1.237287465678552e+00,
         1.285917386547595e+00,  7.822717561930738e-01,
         6.218419925775919e-01,  9.982981161889705e-01,
         1.190509753017736e+00,  8.842681055823868e-01,
         1.209889633346343e+00,  7.175357363544954e-01,
         1.054604687826564e+00,  1.105697974543375e+00,
         1.268017325266862e+00,  7.509008473189935e-01,
         1.168528591982921e+00,  7.685063167283038e-01,
        -2.071014740685577e-01, -5.458350879477755e-01,
         1.505578988885111e-01,  4.439282179604445e-02,
        -2.987476169855480e-01,  1.362998684770691e-02,
        -3.702069362683295e-01, -2.076311052397772e-01,
         2.541618794940901e-01,  1.001740028139758e-01,
         2.455249973556499e-01, -9.933579977845318e-02,
         5.821773524743802e-01, -3.440985876361996e-01,
        -1.107187971891154e-01, -3.208272083210646e-01,
        -3.492925366476388e-01, -1.182884263623090e-01,
         4.384267249862666e-01, -6.000786878777280e-02,
        -5.586495407948650e-02,  2.481276222538253e-01,
         3.424455932619815e-01,  4.767407551309555e-01,
         1.631660171201637e-01, -5.135900732507804e-01,
         2.670948384069483e-01,  6.475513972576710e-01,
         4.043653949084609e-02, -2.098913048058428e-02,
        -2.141768395471972e-01, -1.419352880929378e-01,
        -3.204738826591143e-02, -2.854431682706483e-01,
         3.482403802950401e-01, -2.682525330807729e-01,
         4.379524057248080e-01,  2.555878554029040e-01,
         3.508465461282888e-01, -1.365562289386354e-01,
        -4.835774064766307e-02,  2.889175284456296e-01,
        -7.651902898109114e-02,  4.146722196809335e-01,
         4.035881655568866e-01,  4.096127152326055e-01,
        -3.852926389235675e-02,  1.009436134980639e-01,
    ];

    {
        let mut data = data.clone();
        transform(&mut data, Operation::Forward, &dwt::wavelet::Haar::new(), 6);
        assert::close(&data, &expected_data_6[..], 1e-14);
    }

    {
        let mut data = data.clone();
        transform(&mut data, Operation::Forward, &dwt::wavelet::Haar::new(), 2);
        assert::close(&data, &expected_data_2[..], 1e-14);
    }
}

#[test]
fn inverse_haar_40() {
    let data = vec![
        8.147236863931789e-01, 9.057919370756192e-01, 1.269868162935061e-01, 9.133758561390194e-01,
        6.323592462254095e-01, 9.754040499940952e-02, 2.784982188670484e-01, 5.468815192049838e-01,
        9.575068354342976e-01, 9.648885351992765e-01, 1.576130816775483e-01, 9.705927817606157e-01,
        9.571669482429456e-01, 4.853756487228412e-01, 8.002804688888001e-01, 1.418863386272153e-01,
        4.217612826262750e-01, 9.157355251890671e-01, 7.922073295595544e-01, 9.594924263929030e-01,
        6.557406991565868e-01, 3.571167857418955e-02, 8.491293058687771e-01, 9.339932477575505e-01,
        6.787351548577735e-01, 7.577401305783334e-01, 7.431324681249162e-01, 3.922270195341682e-01,
        6.554778901775566e-01, 1.711866878115618e-01, 7.060460880196088e-01, 3.183284637742068e-02,
        2.769229849608900e-01, 4.617139063115394e-02, 9.713178123584754e-02, 8.234578283272926e-01,
        6.948286229758170e-01, 3.170994800608605e-01, 9.502220488383549e-01, 3.444608050290876e-02,
    ];

    let expected_data_3 = [
         8.650192985407463e-01, -6.233809160651432e-02,
         2.689794918769315e-01,  2.184755517022040e-01,
         1.339284061977924e+00,  1.384338814098467e-01,
         4.286991490050805e-01, -8.921667691385419e-01,
         1.377231504830873e+00,  4.173550435716070e-01,
         4.759295106674139e-01, -5.956768587508252e-01,
         9.899436528629466e-01, -6.100436219910166e-02,
         2.564403818862800e-01, -2.982523886681176e-01,
         1.101881530335739e+00,  1.748958082109356e-01,
        -4.084453181504334e-02, -2.829390674158830e-01,
         4.217379500759987e-01, -5.767620032618002e-01,
        -1.968891436821723e-01, -2.419075867580599e-01,
         1.068151680853419e+00,  6.765234397888883e-01,
         4.832243811069808e-01,  4.179281742827768e-01,
         5.109476464150825e-01,  3.735825640538903e-01,
         1.088021944767958e-01, -1.055743034385957e+00,
         1.452134064588771e+00,  4.694980024513480e-01,
         3.928318966222677e-01, -5.561448870125810e-02,
         1.034087809974746e+00, -3.097290987584052e-01,
        -5.729560136758283e-01, -6.216701278936372e-01,
    ];

    {
        let mut data = data.clone();
        transform(&mut data, Operation::Inverse, &dwt::wavelet::Haar::new(), 3);
        assert::close(&data, &expected_data_3[..], 1e-14);
    }
}

#[test]
fn inverse_haar_64() {
    let data = vec![
        3.516595070629968e-01, 8.308286278962909e-01, 5.852640911527243e-01, 5.497236082911395e-01,
        9.171936638298100e-01, 2.858390188203735e-01, 7.572002291107213e-01, 7.537290942784953e-01,
        3.804458469753567e-01, 5.678216407252211e-01, 7.585428956306361e-02, 5.395011866660715e-02,
        5.307975530089727e-01, 7.791672301020112e-01, 9.340106842291830e-01, 1.299062084737301e-01,
        5.688236608721927e-01, 4.693906410582058e-01, 1.190206950124140e-02, 3.371226443988815e-01,
        1.621823081932428e-01, 7.942845406839070e-01, 3.112150420448049e-01, 5.285331355062127e-01,
        1.656487294997809e-01, 6.019819414016365e-01, 2.629712845401443e-01, 6.540790984767823e-01,
        6.892145031400078e-01, 7.481515928237095e-01, 4.505415985024978e-01, 8.382137799693257e-02,
        2.289769687168188e-01, 9.133373615016696e-01, 1.523780189692230e-01, 8.258169774895474e-01,
        5.383424352600571e-01, 9.961347166268855e-01, 7.817552875318368e-02, 4.426782697754463e-01,
        1.066527701805844e-01, 9.618980808550537e-01, 4.634224134067444e-03, 7.749104647115024e-01,
        8.173032206534330e-01, 8.686947053635097e-01, 8.443584551091032e-02, 3.997826490988965e-01,
        2.598704028506542e-01, 8.000684802243075e-01, 4.314138274635446e-01, 9.106475944295229e-01,
        1.818470283028525e-01, 2.638029165219901e-01, 1.455389803847170e-01, 1.360685587086637e-01,
        8.692922076400893e-01, 5.797045873655702e-01, 5.498602018363320e-01, 1.449547982237268e-01,
        8.530311177218937e-01, 6.220551314850660e-01, 3.509523808922709e-01, 5.132495398670534e-01,
    ];

    let expected_data_6 = [
         1.061401401623698e+00,  7.375790669932929e-01,
         9.764936152651623e-01, -3.151604683925572e-01,
         6.885054166478777e-01,  4.730103556140577e-01,
         6.953080298745993e-01, -4.725735397290757e-01,
         6.093455404763515e-01, -1.519856326693813e-01,
         9.211514975044544e-01, -4.875957286999670e-01,
         4.505815519524989e-02, -6.549873781319032e-02,
        -3.432212926570571e-02, -6.603637421499977e-01,
         2.991343120549443e-01,  1.483045180009010e-01,
         7.417017626176143e-01, -6.186275489482547e-01,
         4.894103400518620e-01,  4.828565574303871e-01,
         2.397933524671397e-01, -8.560955363527047e-01,
         7.254926282645878e-01, -4.303486709547074e-01,
         4.506168535535229e-01, -7.779029803332524e-01,
         2.777876895666517e-01,  1.583773716946829e-01,
        -2.776158269699375e-02, -5.931396270540972e-01,
         6.808280033939725e-01,  3.133157552232174e-01,
         8.971569975890371e-01, -2.343106979714090e-01,
         6.449635789867678e-01,  3.485229319253658e-02,
         3.818510839803482e-01, -9.059990946043170e-01,
         3.835302505817852e-01,  1.263597168786472e-01,
         1.785105303595598e-01, -1.945631319794160e-01,
         2.456058572073344e-03, -2.033671413419453e-01,
        -6.583196392925423e-01, -8.507496404308943e-01,
         1.320870121349718e+00,  9.150529163985532e-02,
         4.268862481659231e-01, -3.929398414563651e-01,
         4.640208402405691e-01, -3.135989146055791e-01,
        -5.704421092166897e-01, -7.754391507957396e-01,
         5.288769311465078e-01, -6.774912446620744e-01,
        -8.498935351530146e-02, -9.647081570052607e-01,
        -1.013640195380850e-01, -5.976856363430628e-01,
        -7.042397585663779e-02, -7.962684360183752e-01,
    ];

    let expected_data_2 = [
         6.221527512827973e-01,  2.983304166523923e-01,
         5.372449649242618e-01, -7.544091187334577e-01,
         7.578571649941583e-01,  5.423621039603383e-01,
         7.646597782208800e-01, -4.032217913827950e-01,
         6.792486668998492e-01, -8.208250624588348e-02,
         9.910546239279522e-01, -4.176926022764692e-01,
         4.987015728492307e-01,  3.881446798407905e-01,
         4.193212883882750e-01, -2.067203244960170e-01,
         6.151028830385481e-01,  4.642730889845048e-01,
         1.057670333601218e+00, -3.026589779646508e-01,
         5.433386710628778e-01,  5.367848884414028e-01,
         2.937216834781554e-01, -8.021672053416889e-01,
         1.112128285187411e+00, -4.371301403188443e-02,
         8.372525104763460e-01, -3.912673234104294e-01,
         7.008362738283385e-01,  5.814259559563697e-01,
         3.952870015646931e-01, -1.700910427924104e-01,
         4.568034123229464e-01,  8.929116415219132e-02,
         6.731324065180110e-01, -4.583352890424351e-01,
         8.899574339605445e-01,  2.798461481663133e-01,
         6.268449389541249e-01, -6.610052396305404e-01,
         2.979980539031730e-01,  4.082752020003494e-02,
         9.297833368094757e-02, -2.800953286580283e-01,
         4.569262085287041e-01,  2.511030086146854e-01,
        -2.038494893359117e-01, -3.962794904742636e-01,
         1.224688442929422e+00, -4.676386780441133e-03,
         3.307045697456266e-01, -4.891215198766616e-01,
         1.152469288885934e+00,  3.748495340397863e-01,
         1.180063394286758e-01, -8.699070215037412e-02,
         1.295460229270132e+00,  8.909205346154936e-02,
         6.815939446083222e-01, -1.981248588816370e-01,
         3.550246016378202e-01, -1.412970151671575e-01,
         3.859646453192675e-01, -3.398798148424699e-01,
    ];

    {
        let mut data = data.clone();
        transform(&mut data, Operation::Inverse, &dwt::wavelet::Haar::new(), 6);
        assert::close(&data, &expected_data_6[..], 1e-14);
    }

    {
        let mut data = data.clone();
        transform(&mut data, Operation::Inverse, &dwt::wavelet::Haar::new(), 2);
        assert::close(&data, &expected_data_2[..], 1e-14);
    }
}