package org.python.pydev.shared_core.structure;

import static org.junit.Assert.assertEquals;

import java.util.List;

import org.junit.Test;

public class TreeNodeTest {

    @Test
    public void test() {
        TreeNode<Integer> root = new TreeNode<Integer>(null, 0);
        TreeNode<Integer> c1 = new TreeNode<Integer>(root, 1);
        TreeNode<Integer> c2 = new TreeNode<Integer>(c1, 2);
        TreeNode<Integer> c3 = new TreeNode<Integer>(c1, 3);
        List<TreeNode<Integer>> flattened = root.flatten();
        assertEquals(flattened.size(), 3);
    }

}

@@ -46,54 +46,8 @@
 e;%0A%0A
-import static org.junit.Assert.assertEquals;%0A%0A
 impo
@@ -77,22 +77,32 @@
 ort 
-org.
 junit.
-Test
+framework.TestCase
 ;%0A%0Ap
@@ -130,20 +130,27 @@
 est 
-%7B%0A%0A    @Test
+extends TestCase %7B%0A
 %0A   
@@ -166,16 +166,24 @@
 oid test
+TreeNode
 () %7B%0A   

Bring TreeNodeTest.java up to JUnit4