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)