FIX: changed methods to public

src/mhtree/LeafNode.java

@@ -12,7 +12,8 @@ import java.util.List;
 /**
  * Represents a leaf node in MH-Tree.
  */
-class LeafNode extends Node implements Serializable {
+
+public class LeafNode extends Node implements Serializable {
 
     /**
      * Serialization ID
@@ -52,7 +53,7 @@ class LeafNode extends Node implements Serializable {
      *
      * @return a list of objects in node's bucket
      */
-    protected List<LocalAbstractObject> getObjects() {
+    public List<LocalAbstractObject> getObjects() {
         List<LocalAbstractObject> objects = new ArrayList<>(bucket.getObjectCount());
 
         bucket

src/mhtree/MHTree.java

@@ -105,13 +105,9 @@ public class MHTree extends Algorithm implements Serializable {
         operation.endOperation();
     }
 
-    private boolean isPrunable(Node child, LocalAbstractObject queryObject, ApproxKNNQueryOperation operation) {
-        return operation.getAnswerDistance() < child.getDistanceToNearest(queryObject);
-    }
-
-    public void kNN(KNNQueryOperation knnQueryOperation) {
-        root.getObjects().forEach(knnQueryOperation::addToAnswer);
-        knnQueryOperation.endOperation();
+    public void kNN(KNNQueryOperation operation) {
+        root.getObjects().forEach(operation::addToAnswer);
+        operation.endOperation();
     }
 
     public int getObjectCount() {
@@ -151,7 +147,7 @@ public class MHTree extends Algorithm implements Serializable {
      *
      * @return a list of leaf nodes
      */
-    private List<LeafNode> getLeafNodes() {
+    public List<LeafNode> getLeafNodes() {
         List<LeafNode> leafNodes = new ArrayList<>();
         root.gatherLeafNodes(leafNodes);
         return leafNodes;
@@ -206,6 +202,10 @@ public class MHTree extends Algorithm implements Serializable {
                 '}';
     }
 
+    private boolean isPrunable(Node child, LocalAbstractObject queryObject, ApproxKNNQueryOperation operation) {
+        return operation.getAnswerDistance() < child.getDistanceToNearest(queryObject);
+    }
+
     public static class Builder {
 
         /**