IsHeightBalancedTest.java

import org.junit.Test;

import static org.junit.Assert.*;

public class IsHeightBalancedTest {

    private boolean expected;
    private BinaryTree<Integer> tree;

    @Test
    public void isBalanced1() throws Exception {
        expected = true;
        tree = BinaryTreeUtil.getEvenTree();

        test(expected, tree);
    }

    @Test
    public void isBalanced2() throws Exception {
        expected = false;
        tree = BinaryTreeUtil.getOddTree();

        test(expected, tree);
    }

    @Test
    public void isBalanced3() throws Exception {
        expected = true;
        tree = BinaryTreeUtil.getFullTree();

        test(expected, tree);
    }

    @Test
    public void isBalanced4() throws Exception {
        expected = true;
        tree = BinaryTreeUtil.getFigureTenDotOne();

        test(expected, tree);
    }

    private void test(boolean expected, BinaryTree<Integer> tree) {
        assertEquals(expected, IsHeightBalanced.isBalanced(tree));
    }

}