Skip to content

Commit 74124ce

Browse files
authored
Added debug module initializations to TestMatrix.cpp
1 parent 8c1e950 commit 74124ce

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/TestMatrix.cpp

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@
1414
#include <functional>
1515
#include <helib/Matrix.h>
1616
#include <helib/helib.h>
17+
#include <helib/debugging.h>
18+
#include <helib/fhe_stats.h>
1719

1820
#include "test_common.h"
1921
#include "gtest/gtest.h"
@@ -196,6 +198,16 @@ class TestMatrixWithCtxt : public ::testing::TestWithParam<BGVParameters>
196198
pk((sk.GenSecKey(), sk)),
197199
ea(*context.ea)
198200
{}
201+
202+
virtual void SetUp() override
203+
{
204+
helib::setupDebugGlobals(&sk, context.ea);
205+
if (!helib_test::noPrint) {
206+
helib::fhe_stats = true;
207+
}
208+
};
209+
210+
virtual void TearDown() override { helib::cleanupDebugGlobals(); }
199211
};
200212

201213
namespace {

0 commit comments

Comments
 (0)