Update tests to 99,1% of coverage

2024-11-23 18:14:15 +01:00
parent 3728bcb7d3
commit 5d01eccf1b
11 changed files with 5108 additions and 206 deletions
--- a/bayesnet/utils/Mst.h
+++ b/bayesnet/utils/Mst.h
@@ -14,6 +14,8 @@ namespace bayesnet {
    public:
        MST() = default;
        MST(const std::vector<std::string>& features, const torch::Tensor& weights, const int root);
+        void insertElement(std::list<int>& variables, int variable);
+        std::vector<std::pair<int, int>> reorder(std::vector<std::pair<float, std::pair<int, int>>> T, int root_original);
        std::vector<std::pair<int, int>> maximumSpanningTree();
    private:
        torch::Tensor weights;