Add parquetAfter() properly in test.

apache · Jul 29, 2014 · de3ae13 · de3ae13
1 parent dcff9bd
commit de3ae13
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/sql/hive/src/test/scala/org/apache/spark/sql/hive/EstimatesSuite.scala b/sql/hive/src/test/scala/org/apache/spark/sql/hive/EstimatesSuite.scala
@@ -100,14 +100,28 @@ class EstimatesSuite extends QueryTest {
       val testRDD = parquetFile(ParquetTestData.testDir.toString)
       testRDD.registerAsTable("psrc")
     }
+    def parquetAfter() = {
+      Utils.deleteRecursively(ParquetTestData.testDir)
+      reset()
+    }
     mkTest(
-     parquetBefore, reset, parquetQuery, parquetAnswer, implicitly[ClassTag[ParquetRelation]])
+      parquetBefore,
+      parquetAfter,
+      parquetQuery,
+      parquetAnswer,
+      implicitly[ClassTag[ParquetRelation]]
+    )
 
     /** Tests for MetastoreRelation */
     val metastoreQuery = """SELECT * FROM src a JOIN src b ON a.key = 238 AND a.key = b.key"""
     val metastoreAnswer = Seq.fill(4)((238, "val_238", 238, "val_238"))
     mkTest(
-      () => (), () => (), metastoreQuery, metastoreAnswer, implicitly[ClassTag[MetastoreRelation]])
+      () => (),
+      () => (),
+      metastoreQuery,
+      metastoreAnswer,
+      implicitly[ClassTag[MetastoreRelation]]
+    )
   }
 
 }