33from tests .utils import TestCaseBase
44
55import sqlparse
6+ from sqlparse .exceptions import SQLParseError
67
78
89class TestFormat (TestCaseBase ):
@@ -15,7 +16,7 @@ def test_keywordcase(self):
1516 self .ndiffAssertEqual (res , 'Select * From bar; -- select foo\n ' )
1617 res = sqlparse .format (sql .upper (), keyword_case = 'lower' )
1718 self .ndiffAssertEqual (res , 'select * from BAR; -- SELECT FOO\n ' )
18- self .assertRaises (sqlparse . SQLParseError , sqlparse .format , sql ,
19+ self .assertRaises (SQLParseError , sqlparse .format , sql ,
1920 keyword_case = 'foo' )
2021
2122 def test_identifiercase (self ):
@@ -26,7 +27,7 @@ def test_identifiercase(self):
2627 self .ndiffAssertEqual (res , 'select * from Bar; -- select foo\n ' )
2728 res = sqlparse .format (sql .upper (), identifier_case = 'lower' )
2829 self .ndiffAssertEqual (res , 'SELECT * FROM bar; -- SELECT FOO\n ' )
29- self .assertRaises (sqlparse . SQLParseError , sqlparse .format , sql ,
30+ self .assertRaises (SQLParseError , sqlparse .format , sql ,
3031 identifier_case = 'foo' )
3132 sql = 'select * from "foo"."bar"'
3233 res = sqlparse .format (sql , identifier_case = "upper" )
@@ -42,7 +43,7 @@ def test_strip_comments_single(self):
4243 sql = 'select-- foo\n from -- bar\n where'
4344 res = sqlparse .format (sql , strip_comments = True )
4445 self .ndiffAssertEqual (res , 'select from where' )
45- self .assertRaises (sqlparse . SQLParseError , sqlparse .format , sql ,
46+ self .assertRaises (SQLParseError , sqlparse .format , sql ,
4647 strip_comments = None )
4748
4849 def test_strip_comments_multi (self ):
@@ -65,7 +66,7 @@ def test_strip_ws(self):
6566 self .ndiffAssertEqual (f (s ), 'select * from foo where (1 = 2)' )
6667 s = 'select -- foo\n from bar\n '
6768 self .ndiffAssertEqual (f (s ), 'select -- foo\n from bar' )
68- self .assertRaises (sqlparse . SQLParseError , sqlparse .format , s ,
69+ self .assertRaises (SQLParseError , sqlparse .format , s ,
6970 strip_whitespace = None )
7071
7172 def test_preserve_ws (self ):
@@ -76,20 +77,20 @@ def test_preserve_ws(self):
7677
7778 def test_outputformat (self ):
7879 sql = 'select * from foo;'
79- self .assertRaises (sqlparse . SQLParseError , sqlparse .format , sql ,
80+ self .assertRaises (SQLParseError , sqlparse .format , sql ,
8081 output_format = 'foo' )
8182
8283
8384class TestFormatReindent (TestCaseBase ):
8485
8586 def test_option (self ):
86- self .assertRaises (sqlparse . SQLParseError , sqlparse .format , 'foo' ,
87+ self .assertRaises (SQLParseError , sqlparse .format , 'foo' ,
8788 reindent = 2 )
88- self .assertRaises (sqlparse . SQLParseError , sqlparse .format , 'foo' ,
89+ self .assertRaises (SQLParseError , sqlparse .format , 'foo' ,
8990 indent_tabs = 2 )
90- self .assertRaises (sqlparse . SQLParseError , sqlparse .format , 'foo' ,
91+ self .assertRaises (SQLParseError , sqlparse .format , 'foo' ,
9192 reindent = True , indent_width = 'foo' )
92- self .assertRaises (sqlparse . SQLParseError , sqlparse .format , 'foo' ,
93+ self .assertRaises (SQLParseError , sqlparse .format , 'foo' ,
9394 reindent = True , indent_width = - 12 )
9495
9596 def test_stmts (self ):
0 commit comments