package cs6375.github.io;

import org.junit.Test;

import java.io.IOException;
import static org.junit.Assert.*;

public class DataSetTest {
    @Test
    public void split() throws Exception {
        DataSet ds = new DataSet("data/data_sets1/training_set.csv");
        DataSet a = new DataSet(ds.getAttributes(), ds.getDimension());
        DataSet b = new DataSet(ds.getAttributes(), ds.getDimension());
        ds.split(a, b, "XH");
        System.out.println(a);
        System.out.println(b);
        System.out.println(ds.getNumOfInstance());
    }

    @Test
    public void DataSet() throws IOException {
        DataSet ds = new DataSet("data/data_sets1/training_set.csv");
        System.out.println(ds);
    }

}
@@ -199,67 +199,163 @@
     
-DataSet ds = new DataSet(%22data/data_sets1/training_set.csv%22
+String trainingfile = this.getClass().getClassLoader().getResource(%22data_sets1/training_set.csv%22).getFile();%0A%0A        DataSet ds = new DataSet(trainingfile
 );%0A 
@@ -651,171 +651,5 @@
  %7D%0A%0A
-    @Test%0A    public void DataSet() throws IOException %7B%0A        DataSet ds = new DataSet(%22data/data_sets1/training_set.csv%22);%0A        System.out.println(ds);%0A    %7D%0A%0A
 %7D

Update gradle file.