Use case clauses and switch statement instead of is/asInstanceOf

1 files changed, 4 insertions, 2 deletions

diff --git a/tests/shared/src/main/scala/magnolia/main.scala b/tests/shared/src/main/scala/magnolia/main.scala
index 9449f8d..3ac3fb6 100644
--- a/tests/shared/src/main/scala/magnolia/main.scala
+++ b/tests/shared/src/main/scala/magnolia/main.scala
@@ -5,13 +5,15 @@ import examples._
 object Main {
   def main(args: Array[String]): Unit = {
 
-    val tree1: Tree = Branch(Branch(Leaf(1, "abc"), Leaf(2, "def")), Leaf(3, "ghi"))
-    val tree2: Tree = Branch(Leaf(1, "abc"), Leaf(2, "def"))
+    val tree1: Tree = Branch(Branch(Leaf(1), Leaf(2)), Leaf(3))
+    val tree2: Tree = Branch(Leaf(1), Leaf(2))
 
     println(tree1.show)
     println(tree1 isEqualTo tree1)
     println(tree1 isEqualTo tree2)
 
+    println(Branch(Branch(Leaf(1), Leaf(2)), Leaf(3)).show)
+    
     println(List[Entity](Person("John Smith",
       Address(List("1 High Street", "London", "SW1A 1AA"),
         Country("UK", "GBR", false)))).show)