v1.2.2

README.md

@@ -22,7 +22,7 @@ val poland = TreeNode("Poland")
 val france = TreeNode("France")
 europe.addChild(poland)
 europe.addChild(france)
-println(root)
+println(root.prettyString())
 ```
 
 **Pretty Kotlin**
@@ -55,7 +55,7 @@ TreeNode<String> poland = new TreeNode<>("Poland");
 TreeNode<String> france = new TreeNode<>("France");
 europe.addChild(poland);
 europe.addChild(france);
-System.out.println(root);
+System.out.println(root.prettyString());
 ```
 
 *Output:*

treedatastructure/build.gradle

@@ -15,7 +15,7 @@ android {
         minSdkVersion 15
         targetSdkVersion 29
         versionCode 1
-        versionName "1.2.0"
+        versionName "1.2.2"
 
         testInstrumentationRunner "androidx.test.runner.AndroidJUnitRunner"
         consumerProguardFiles 'consumer-rules.pro'

treedatastructure/src/main/java/com/github/adriankuta/datastructure/tree/TreeNode.kt

@@ -27,7 +27,7 @@ open class TreeNode<T>(val value: T) : Iterable<TreeNode<T>>, ChildDeclarationIn
     }
 
     @JvmSynthetic
-    override fun child(child: T, childDeclaration: ChildDeclaration<T>?) {
+    override fun child(value: T, childDeclaration: ChildDeclaration<T>?) {
         val newChild = TreeNode(value)
         if(childDeclaration != null)
             newChild.childDeclaration()
@@ -91,6 +91,10 @@ open class TreeNode<T>(val value: T) : Iterable<TreeNode<T>>, ChildDeclarationIn
     }
 
     override fun toString(): String {
+        return value.toString()
+    }
+
+    fun prettyString(): String {
         val stringBuilder = StringBuilder()
         print(stringBuilder, "", "")
         return stringBuilder.toString()

treedatastructure/src/test/java/com/github/adriankuta/datastructure/tree/TreeNodeTest.kt

@@ -128,18 +128,31 @@ class TreeNodeTest {
         europe.addChild(poland)
         europe.addChild(france)
 
-        val rootExt = treeNode("World") {
+        val rootExt = tree("World") {
-            treeNode("North America") {
+            child("North America") {
-                treeNode("USA")
+                child("USA")
             }
-            treeNode("Europe") {
+            child("Europe") {
-                treeNode("Poland")
+                child("Poland")
-                treeNode("France")
+                child("France")
             }
         }
-        println(root)
+        assertEquals(root.prettyString(), rootExt.prettyString())
-        assertEquals(root.toString(), rootExt.toString())
     }
 
+//    @Test
+//    fun getVisibleNodes() {
+//        val root = tree("World") {
+//            child("Level 1") {
+//                child("Level 2") {
+//                    child("Level 3") {
+//                        child("Level 4")
+//                    }
+//                }
+//            }
+//        }
+//
+//        print(root.prettyString())
+//    }
 
 }