import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;

import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;


public class SimpleTest {

    public static Partitioner partitioner;

    @BeforeClass
    public static void init() {
        partitioner = new MyPartitioner();
    }

    @Test
    public void test1() {
        List<Partition> antw = partitioner.partitions(3);
        Set<Partition> antwSet = new HashSet<>(antw);
        Assert.assertTrue("Je hebt sommige partities meermaals gegenereerd!", antw.size() == antwSet.size());
        Assert.assertEquals(new HashSet<>(Arrays.asList(
                new Partition(3),
                new Partition(2, 1),
                new Partition(1, 1, 1)
        )), antwSet);
    }

    @Test
    public void test2() {
        List<Partition> antw = partitioner.partitions(5);
        Set<Partition> antwSet = new HashSet<>(antw);
        Assert.assertTrue("Je hebt sommige partities meermaals gegenereerd!", antw.size() == antwSet.size());
        Assert.assertEquals(new HashSet<>(Arrays.asList(
                new Partition(5),
                new Partition(4, 1),
                new Partition(3, 2),
                new Partition(3, 1, 1),
                new Partition(2, 2, 1),
                new Partition(2, 1, 1, 1),
                new Partition(1, 1, 1, 1, 1)
        )), antwSet);
    }

    @Test
    public void test3() {
        List<Partition> antw = partitioner.partitions(6);
        Set<Partition> antwSet = new HashSet<>(antw);
        Assert.assertTrue("Je hebt sommige partities meermaals gegenereerd!", antw.size() == antwSet.size());
        Assert.assertEquals(new HashSet<>(Arrays.asList(
                new Partition(6),
                new Partition(5, 1),
                new Partition(4, 2),
                new Partition(4, 1, 1),
                new Partition(3, 3),
                new Partition(3, 2, 1),
                new Partition(3, 1, 1, 1),
                new Partition(2, 2, 2),
                new Partition(2, 2, 1, 1),
                new Partition(2, 1, 1, 1, 1),
                new Partition(1, 1, 1, 1, 1, 1)
        )), antwSet);
    }

    @Test
    public void test4() {
        List<Partition> antw = partitioner.partitions(2);
        Set<Partition> antwSet = new HashSet<>(antw);
        Assert.assertTrue("Je hebt sommige partities meermaals gegenereerd!", antw.size() == antwSet.size());
        Assert.assertEquals(new HashSet<>(Arrays.asList(
                new Partition(2),
                new Partition(1, 1)
        )), antwSet);
    }

    @Test(expected = IllegalArgumentException.class)
    public void testNegatiefGetal() {
        List<Partition> antw = partitioner.partitions(-4);
    }

    @Test(expected = IllegalArgumentException.class)
    public void testNul() {
        List<Partition> antw = partitioner.partitions(0);
    }
    
}