import static org.junit.Assert.*;

import org.junit.Test;


public class FlowerParserTest {


	// .data file - not enough features in a line
	@Test (expected = IllegalArgumentException.class)
	public void testNotEnoughFeatures() {
		Flower[] flowers = FlowerParser.parse("notEnoughFeatures.data");
	}

	@Test (expected = IllegalArgumentException.class)
	public void testFileNotFound() {
		Flower[] flowers = FlowerParser.parse("doesNotExist.data");
	}

	@Test (expected = IllegalArgumentException.class)
	public void testFileNull() {
		Flower[] flowers = FlowerParser.parse(null);
	}

	// . data file - some of the features are not doubles
	@Test (expected = IllegalArgumentException.class)
	public void testFeaturesNotDoubles() {
		Flower[] flowers = FlowerParser.parse("notDoubles.data");
	}


	@Test
    public void testIrisTrainingSet() {
            Flower[] flowers = FlowerParser.parse("irisTrainingSet.data");

            assertTrue(flowers[0].getFeatures()[0] == 5.4);
            assertTrue(flowers[0].getFeatures()[1] == 3.7);
            assertTrue(flowers[0].getFeatures()[2] == 1.5);
            assertTrue(flowers[0].getFeatures()[3] == 0.2);
            assertTrue(flowers[0].label.equals("Iris-setosa"));

            assertTrue(flowers[1].getFeatures()[0] == 4.8);
            assertTrue(flowers[1].getFeatures()[1] == 3.4);
            assertTrue(flowers[1].getFeatures()[2] == 1.6);
            assertTrue(flowers[1].getFeatures()[3] == 0.2);
            assertTrue(flowers[1].label.equals("Iris-setosa"));

            assertTrue(flowers[flowers.length-1].getFeatures()[0] == 6.9);
            assertTrue(flowers[flowers.length-1].getFeatures()[1] == 3.1);
            assertTrue(flowers[flowers.length-1].getFeatures()[2] == 5.4);
            assertTrue(flowers[flowers.length-1].getFeatures()[3] == 2.1);
            assertTrue(flowers[flowers.length-1].label.equals("Iris-virginica"));

            assertTrue(flowers.length == 120);
    }

    @Test
    public void testIrisTestSet() {
            Flower[] flowers = FlowerParser.parse("irisTestSet.data");

            assertTrue(flowers[0].getFeatures()[0] == 5.1);
            assertTrue(flowers[0].getFeatures()[1] == 3.5);
            assertTrue(flowers[0].getFeatures()[2] == 1.4);
            assertTrue(flowers[0].getFeatures()[3] == 0.2);
            assertTrue(flowers[0].label.equals("N/A"));

            assertTrue(flowers[1].getFeatures()[0] == 4.9);
            assertTrue(flowers[1].getFeatures()[1] == 3.0);
            assertTrue(flowers[1].getFeatures()[2] == 1.4);
            assertTrue(flowers[1].getFeatures()[3] == 0.2);
            assertTrue(flowers[1].label.equals("N/A"));

            assertTrue(flowers[flowers.length-1].getFeatures()[0] == 5.9);
            assertTrue(flowers[flowers.length-1].getFeatures()[1] == 3.0);
            assertTrue(flowers[flowers.length-1].getFeatures()[2] == 5.1);
            assertTrue(flowers[flowers.length-1].getFeatures()[3] == 1.8);
            assertTrue(flowers[flowers.length-1].label.equals("N/A"));

            assertTrue(flowers.length == 30);
    }




}