[wip] use native annotation tags

.github/workflows/test_scala_and_python.yaml

@@ -84,7 +84,7 @@ jobs:
       - name: Run other spark tests
         run: |
           export SBT_OPTS="-Xmx8G -Xms2G --add-opens=java.base/sun.nio.ch=ALL-UNNAMED"
-          sbt "spark/testOnly -- -n jointest"
+          sbt "spark/testOnly -- -n ai.chronon.spark.test.JoinTestTag"
 
   mutation_spark_tests:
     runs-on: ubuntu-latest
@@ -103,7 +103,7 @@ jobs:
       - name: Run other spark tests
         run: |
           export SBT_OPTS="-Xmx8G -Xms2G --add-opens=java.base/sun.nio.ch=ALL-UNNAMED"
-          sbt "spark/testOnly -- -n mutationstest"
+          sbt "spark/testOnly -- -n ai.chronon.spark.test.MutationsTestTag"
 
   fetcher_spark_tests:
     runs-on: ubuntu-latest
@@ -122,7 +122,7 @@ jobs:
       - name: Run other spark tests
         run: |
           export SBT_OPTS="-Xmx8G -Xms2G --add-opens=java.base/sun.nio.ch=ALL-UNNAMED"
-          sbt "spark/testOnly -- -n fetchertest"
+          sbt "spark/testOnly -- -n ai.chronon.spark.test.FetcherTestTag"
 
   scala_compile_fmt_fix      :
     runs-on: ubuntu-latest

spark/src/test/java/ai/chronon/spark/test/Tags.java

@@ -0,0 +1,23 @@
+package ai.chronon.spark.test;
+
+import java.lang.annotation.Retention;
+import java.lang.annotation.Target;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.ElementType;
+
+@org.scalatest.TagAnnotation
+@Retention(RetentionPolicy.RUNTIME)
+@Target({ElementType.METHOD, ElementType.TYPE})
+@interface JoinTestTag {}
+
+
+@org.scalatest.TagAnnotation
+@Retention(RetentionPolicy.RUNTIME)
+@Target({ElementType.METHOD, ElementType.TYPE})
+@interface FetcherTestTag {}
+
+
+@org.scalatest.TagAnnotation
+@Retention(RetentionPolicy.RUNTIME)
+@Target({ElementType.METHOD, ElementType.TYPE})
+@interface MutationsTestTag {}

spark/src/test/scala/ai/chronon/spark/test/FetcherTest.scala

@@ -64,10 +64,9 @@ import scala.concurrent.duration.SECONDS
 import scala.io.Source
 import scala.util.ScalaJavaConversions._
 
-// Run as follows: sbt "spark/testOnly -- -n fetchertest"
-class FetcherTest extends AnyFunSuite with TaggedFilterSuite {
-
-  override def tagName: String = "fetchertest"
+// Run as follows: sbt "spark/testOnly -- -n ai.chronon.spark.test.FetcherTestTag"
+@FetcherTestTag
+class FetcherTest extends AnyFunSuite {
 
   @transient lazy val logger: Logger = LoggerFactory.getLogger(getClass)
   val sessionName = "FetcherTest"

spark/src/test/scala/ai/chronon/spark/test/JoinTest.scala

@@ -40,12 +40,14 @@ import org.apache.spark.sql.types._
 import org.apache.spark.sql.types.{StringType => SparkStringType}
 import org.junit.Assert._
 import org.scalatest.funsuite.AnyFunSuite
+import org.scalatest.{Ignore, Tag, TagAnnotation}
 
 import scala.collection.JavaConverters._
 import scala.util.ScalaJavaConversions.ListOps
 
-// Run as follows: sbt "spark/testOnly -- -n jointest"
-class JoinTest extends AnyFunSuite with TaggedFilterSuite {
+// Run as follows: sbt "spark/testOnly -- -n ai.chronon.spark.test.JoinTestTag"
+@JoinTestTag
+class JoinTest extends AnyFunSuite {
 
   val spark: SparkSession = SparkSessionBuilder.build("JoinTest", local = true)
   private implicit val tableUtils = TableUtils(spark)
@@ -58,8 +60,6 @@ class JoinTest extends AnyFunSuite with TaggedFilterSuite {
   private val namespace = "test_namespace_jointest"
   tableUtils.createDatabase(namespace)
 
-  override def tagName: String = "jointest"
-
   test("test events entities snapshot") {
     val dollarTransactions = List(
       Column("user", StringType, 100),

spark/src/test/scala/ai/chronon/spark/test/MutationsTest.scala

@@ -46,12 +46,12 @@ import org.slf4j.LoggerFactory
   * Left is an event source with definite ts.
   * Right is an entity with snapshots and mutation values through the day.
   * Join is the events and the entity value at the exact timestamp of the ts.
-  *  To run: sbt "spark/testOnly -- -n mutationstest"
+  *  To run: sbt "spark/testOnly -- -n ai.chronon.spark.test.MutationsTestTag"
   */
-class MutationsTest extends AnyFunSuite with TaggedFilterSuite {
-  @transient lazy val logger: Logger = LoggerFactory.getLogger(getClass)
 
-  override def tagName: String = "mutationstest"
+@MutationsTestTag
+class MutationsTest extends AnyFunSuite {
+  @transient lazy val logger: Logger = LoggerFactory.getLogger(getClass)
 
   val spark: SparkSession =
     SparkSessionBuilder.build("MutationsTest",