Added more test cases for select with limit.

amplab · Jul 17, 2013 · 005ce06 · 005ce06
1 parent e505770
commit 005ce06
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/test/scala/shark/SQLSuite.scala b/src/test/scala/shark/SQLSuite.scala
@@ -123,8 +123,13 @@ class SQLSuite extends FunSuite with BeforeAndAfterAll {
   }
 
   test("limit") {
-    assert(sc.sql("select * from test limit 10").length == 10)
-    assert(sc.sql("select * from test limit 501").length == 500)
+    assert(sc.sql("select * from test limit 10").length === 10)
+    assert(sc.sql("select * from test limit 501").length === 500)
+    sc.sql("drop table if exists test_limit0")
+    assert(sc.sql("select * from test limit 0").length === 0)
+    assert(sc.sql("create table test_limit0 as select * from test limit 0").length === 0)
+    assert(sc.sql("select * from test_limit0 limit 0").length === 0)
+    assert(sc.sql("select * from test_limit0 limit 1").length === 0)
   }
 
   //////////////////////////////////////////////////////////////////////////////