Fixed some tests

1 files changed, 15 insertions, 3 deletions

diff --git a/tests/test_functions.py b/tests/test_functions.py
index d5d8f91..5ab1586 100644
--- a/tests/test_functions.py
+++ b/tests/test_functions.py
@@ -63,17 +63,29 @@ LIMIT 1"""
 
 class Test_Compact(Test_SQL):
     def test_compact1(self):
-        self.assertEqual(Tokens2Unicode(compact(self.sql, 'tests/files')),
+        stream = compact(tokenize(self.sql), 'tests/files')
+
+        result = Tokens2Unicode(stream)
+
+        self.assertEqual(result,
             'INSERT INTO dir_entries(type)VALUES(:type);INSERT INTO '
             'directories(inode)VALUES(:inode)LIMIT 1')
 
     def test_compact2(self):
-        self.assertEqual(Tokens2Unicode(compact(self.sql2)),
+        stream = tokenize(self.sql2)
+
+        result = compact(stream)
+
+        self.assertEqual(Tokens2Unicode(result),
             'SELECT child_entry,asdf AS inode,creation FROM links WHERE '
             'parent_dir==:parent_dir AND name==:name LIMIT 1')
 
     def test_compact3(self):
-        self.assertEqual(Tokens2Unicode(compact(self.sql3)),
+        stream = tokenize(self.sql3)
+
+        result = compact(stream)
+
+        self.assertEqual(Tokens2Unicode(result),
             'SELECT 0 AS st_dev,0 AS st_uid,0 AS st_gid,dir_entries.type AS '
             'st_mode,dir_entries.inode AS st_ino,COUNT(links.child_entry)AS '
             'st_nlink,:creation AS st_ctime,dir_entries.access AS st_atime,'