<gh_stars>1-10
import unittest

import atc


class TestSparkImport(unittest.TestCase):
    def test_spark(self):
        spark = atc.spark.Spark.master("local[*]").get()
        df = spark.sql("select 42")
        self.assertEqual(42, df.collect()[0][0])


if __name__ == "__main__":
    unittest.main()

import unittest

import atc


class TestSparkImport(unittest.TestCase):
    def test_spark(self):
        spark = atc.spark.Spark.master("local[*]").get()
        df = spark.sql("select 42")
        self.assertEqual(42, df.collect()[0][0])


if __name__ == "__main__":
    unittest.main()