def test_is_star_table(self):
df = self.spark.createDataFrame([('a', 1), ('b', 2), ('c', 3)], ["key", "value"])
df.write.format("parquet").save(self.tempFile)
tempFile2 = self.tempFile + '_2'
df.write.format("star").save(tempFile2)
self.assertEqual(StarTable.isStarTable(self.spark, self.tempFile), False)
self.assertEqual(StarTable.isStarTable(self.spark, tempFile2), True)
def test_drop_table(self):
self.__writeStarTable([('a', 1), ('b', 2), ('c', 3)])
StarTable.forPath(self.spark, self.tempFile).dropTable()
self.assertEqual(StarTable.isStarTable(self.spark, self.tempFile), False)