rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 6 Wiki Activity

Add getNumberOfNodes & getNumberOfEdges to Models

Browse Source 
Add some more tests
This commit is contained in:
Ricardo Montañana Gómez
2023-07-19 15:05:44 +02:00
parent 1a21015492
commit 2f5bd0ea7e
8 changed files with 66 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/BayesNetwork.cc
View File

@@ -7,7 +7,7 @@
TEST_CASE("Test Bayesian Network")
{
auto[Xd, y, features, className, states] = loadFile("iris");
auto [Xd, y, features, className, states] = loadFile("iris");
SECTION("Test Update Nodes")
{
@@ -26,4 +26,16 @@ TEST_CASE("Test Bayesian Network")
net.addNode("C", 2);
REQUIRE(net.getFeatures() == vector<string>{"A", "B", "C"});
}
SECTION("Test get edges")
{
auto net = bayesnet::Network();
net.addNode("A", 3);
net.addNode("B", 5);
net.addNode("C", 2);
net.addEdge("A", "B");
net.addEdge("B", "C");
REQUIRE(net.getEdges() == vector<pair<string, string>>{ {"A", "B"}, { "B", "C" } });
net.addEdge("A", "C");
REQUIRE(net.getEdges() == vector<pair<string, string>>{ {"A", "B"}, { "A", "C" }, { "B", "C" } });
}
}