Allow whitespaces and comments after the last statement

[sqlconvert.git] / tests / test_stgrouper.py

diff --git a/tests/test_stgrouper.py b/tests/test_stgrouper.py
index 0e72ced759e47ecad9130d1012c05d84a53ba1e4..36a14c2c6350b7ff76b86e63716ac7e76ed7e711 100755 (executable)
--- a/tests/test_stgrouper.py
+++ b/tests/test_stgrouper.py
@@ -31,7 +31,7 @@ class TestStGrouper(unittest.TestCase):
         self.assertEqual(query, 'SELECT * FROM "T";')
         self.assertRaises(StopIteration, next, g)
         self.assertEqual(len(grouper.statements), 0)
-        self.assertIsNone(grouper.close())
+        self.assertEqual(grouper.close(), [])
 
 if __name__ == "__main__":
     main()