relocate

Co-authored-by: Thomas Chow <[email protected]>

Co-authored-by: Thomas Chow <[email protected]>

Author: tchow-zlai

spark/BUILD.bazel

@@ -202,16 +202,6 @@ scala_test_suite(
     deps = test_deps + [":test_lib"],
 )
 
-scala_test_suite(
-    name = "bootstrap_test",
-    srcs = glob(["src/test/scala/ai/chronon/spark/test/bootstrap/*.scala"]),
-    data = glob(["spark/src/test/resources/**/*"]),
-    # defined in prelude_bazel file
-    jvm_flags = _JVM_FLAGS_FOR_ACCESSING_BASE_JAVA_CLASSES,
-    visibility = ["//visibility:public"],
-    deps = test_deps + [":test_lib"],
-)
-
 jvm_binary(
     name = "spark_assembly",
     deploy_env = ["//tools/build_rules/spark:spark"],

spark/src/test/scala/ai/chronon/spark/test/MutationsTest.scala renamed to spark/src/test/scala/ai/chronon/spark/test/join/MutationsTest.scala

@@ -14,33 +14,19 @@
  *    limitations under the License.
  */
 
-package ai.chronon.spark.test
+package ai.chronon.spark.test.join
 
 import ai.chronon.aggregator.test.Column
 import ai.chronon.aggregator.windowing.TsUtils
 import ai.chronon.api
-import ai.chronon.api.Builders
-import ai.chronon.api.Operation
-import ai.chronon.api.TimeUnit
-import ai.chronon.api.Window
-import ai.chronon.spark.Comparison
+import ai.chronon.api.{Builders, Operation, TimeUnit, Window}
 import ai.chronon.spark.Extensions._
-import ai.chronon.spark.Join
-import ai.chronon.spark.SparkSessionBuilder
-import ai.chronon.spark.TableUtils
-import org.apache.spark.sql.DataFrame
-import org.apache.spark.sql.Row
-import org.apache.spark.sql.SparkSession
-import org.apache.spark.sql.types.BooleanType
-import org.apache.spark.sql.types.DoubleType
-import org.apache.spark.sql.types.IntegerType
-import org.apache.spark.sql.types.LongType
-import org.apache.spark.sql.types.StringType
-import org.apache.spark.sql.types.StructField
-import org.apache.spark.sql.types.StructType
+import ai.chronon.spark.{Comparison, Join, SparkSessionBuilder, TableUtils}
+import ai.chronon.spark.test.DataFrameGen
+import org.apache.spark.sql.{DataFrame, Row, SparkSession}
+import org.apache.spark.sql.types._
 import org.scalatest.flatspec.AnyFlatSpec
-import org.slf4j.Logger
-import org.slf4j.LoggerFactory
+import org.slf4j.{Logger, LoggerFactory}
 
 /** Tests for the temporal join of entities.
   * Left is an event source with definite ts.