apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bug fix

Browse Source
This commit is contained in:
heisenbuug
2021-11-10 12:16:48 +05:30
parent 0d736d340a
commit e15eea6bf7
1 changed files with 21 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/tests/digamma_test.cpp
+21 -3
View File
@@ -16,7 +16,7 @@
#include "catch.hpp"
using namespace mlpack;
using namespace arma;
using namespace math;
/**
* Test the output of digamma for large input values.
@@ -25,10 +25,10 @@ TEST_CASE("DigammaLarge", "[DigammaTest]")
{
arma::mat data;
if (!data::Load("digamma_data.csv", data));
if (!data::Load("digamma_data.csv", data))
FAIL("Cannot load data digamma_data.csv");
for (int i = 0; i < data.n_rows; i++)
for (size_t i = 0; i < data.n_rows; i++)
REQUIRE(Digamma(data(i, 0) == Approx(data(i, 1).epsilon(1e-7)));
}
@@ -37,7 +37,13 @@ TEST_CASE("DigammaLarge", "[DigammaTest]")
*/
TEST_CASE("DigammaNegative", "[DigammaTest]")
{
arma::mat data;
if (!data::Load("digamma_neg_data.csv", data))
FAIL("Cannot load data digamma_neg_data.csv");
for (size_t i = 0; i < data.n_rows; i++)
REQUIRE(Digamma(data(i, 0) == Approx(data(i, 1).epsilon(1e-7)));
}
/**
@@ -45,7 +51,13 @@ TEST_CASE("DigammaNegative", "[DigammaTest]")
*/
TEST_CASE("DigammaSmall", "[DigammaTest]")
{
arma::mat data;
if (!data::Load("digamma_small_data.csv", data))
FAIL("Cannot load data digamma_small_data.csv");
for (size_t i = 0; i < data.n_rows; i++)
REQUIRE(Digamma(data(i, 0) == Approx(data(i, 1).epsilon(1e-7)));
}
/**
@@ -53,5 +65,11 @@ TEST_CASE("DigammaSmall", "[DigammaTest]")
*/
TEST_CASE("DigammaNearPositiveRoots", "[DigammaTest]")
{
arma::mat data;
if (!data::Load("digamma_root_data.csv", data))
FAIL("Cannot load data digamma_root_data.csv");
for (size_t i = 0; i < data.n_rows; i++)
REQUIRE(Digamma(data(i, 0) == Approx(data(i, 1).epsilon(1e-7)));
}