package org.deeplearning4j.datasets.iterator;

import org.nd4j.linalg.dataset.DataSet;
import org.junit.Test;
import org.springframework.core.io.ClassPathResource;

/**
 * CSV Test
 * @author Adam Gibson
 */
public class CSVDataSetIteratorTest {
    @Test
    public void testCSV() throws Exception {
       DataSetIterator iter = new CSVDataSetIterator(10,10,new ClassPathResource("csv-example.csv").getFile(),1,1);
        DataSet next = iter.next();
    }


}

@@ -156,16 +156,50 @@
 esource;
+%0Aimport static org.junit.Assert.*;
 %0A%0A/**%0A *
@@ -357,16 +357,27 @@
 r iter =
+%0A          
  new CSV
@@ -395,19 +395,21 @@
 ator(10,
+ 
 10,
+ 
 new Clas
@@ -455,10 +455,12 @@
 e(),
+ 
 1,
+ 
 1);%0A
@@ -462,25 +462,24 @@
  1);%0A       
- 
 DataSet next
@@ -494,16 +494,113 @@
 next();%0A
+       assertEquals(%22%22, 10, next.numExamples());%0A       assertEquals(%22%22, 479, next.numInputs());%0A
     %7D%0A%0A%0A

Make testCSV actually test something