package com.github.dkanellis.skyspark.api.test_utils.base;

import org.apache.spark.api.java.JavaPairRDD;
import org.apache.spark.api.java.JavaRDD;
import org.apache.spark.api.java.JavaSparkContext;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import scala.Tuple2;

import java.util.List;

public abstract class BaseSparkTest {

    private static JavaSparkContext sparkContext;

    @BeforeClass
    public static void setUpClass() {
        sparkContext = new JavaSparkContext("local[*]", "Tests");
    }

    @AfterClass
    public static void tearDownClass() {
        sparkContext.stop();
    }

    protected static JavaSparkContext getSparkContext() {
        return sparkContext;
    }

    protected <T> JavaRDD<T> toRdd(List<T> listOfElements) {
        return sparkContext.parallelize(listOfElements);
    }

    protected <K, V> JavaPairRDD<K, V> toPairRdd(List<Tuple2<K, V>> pairs) {
        return sparkContext.parallelizePairs(pairs);
    }
}

@@ -59,98 +59,8 @@
 %0D%0A%0D%0A
-import org.apache.spark.api.java.JavaPairRDD;%0D%0Aimport org.apache.spark.api.java.JavaRDD;%0D%0A
 impo
@@ -107,16 +107,16 @@
 ntext;%0D%0A
+
 import o
@@ -171,56 +171,8 @@
 ss;%0D
-%0Aimport scala.Tuple2;%0D%0A%0D%0Aimport java.util.List;%0D
 %0A%0D%0Ap
@@ -550,24 +550,24 @@
 ontext() %7B%0D%0A
+
         retu
@@ -594,278 +594,8 @@
   %7D%0D
-%0A%0D%0A    protected %3CT%3E JavaRDD%3CT%3E toRdd(List%3CT%3E listOfElements) %7B%0D%0A        return sparkContext.parallelize(listOfElements);%0D%0A    %7D%0D%0A%0D%0A    protected %3CK, V%3E JavaPairRDD%3CK, V%3E toPairRdd(List%3CTuple2%3CK, V%3E%3E pairs) %7B%0D%0A        return sparkContext.parallelizePairs(pairs);%0D%0A    %7D%0D
 %0A%7D%0D%0A

Remove extra methods