Add some tests

tests/BayesNetwork.cc

@@ -1,3 +1,29 @@
-//
-// Created by Ricardo Montañana Gómez on 18/7/23.
-//
+#include <catch2/catch_test_macros.hpp>
+#include <catch2/catch_approx.hpp>
+#include <catch2/generators/catch_generators.hpp>
+#include <string>
+#include "../src/KDB.h"
+#include "utils.h"
+
+TEST_CASE("Test Bayesian Network")
+{
+    auto[Xd, y, features, className, states] = loadFile("iris");
+
+    SECTION("Test Update Nodes")
+    {
+        auto net = bayesnet::Network();
+        net.addNode("A", 3);
+        REQUIRE(net.getStates() == 3);
+        net.addNode("A", 5);
+        REQUIRE(net.getStates() == 5);
+    }
+    SECTION("Test get features")
+    {
+        auto net = bayesnet::Network();
+        net.addNode("A", 3);
+        net.addNode("B", 5);
+        REQUIRE(net.getFeatures() == vector<string>{"A", "B"});
+        net.addNode("C", 2);
+        REQUIRE(net.getFeatures() == vector<string>{"A", "B", "C"});
+    }
+}