fix to use specific imports

scgkiran · Sep 12, 2024 · 8ec50b5 · 8ec50b5
1 parent 39ff792
commit 8ec50b5
Show file tree

Hide file tree

Showing 6 changed files with 30 additions and 9 deletions.
diff --git a/tests/coverage/antlr_parser/FuncTestCaseLexer.py b/tests/coverage/antlr_parser/FuncTestCaseLexer.py
@@ -1,6 +1,11 @@
 # Generated from FuncTestCaseLexer.g4 by ANTLR 4.13.2
-from antlr4 import *
-from io import StringIO
+from antlr4 import (
+    ATNDeserializer,
+    DFA,
+    Lexer,
+    LexerATNSimulator,
+    PredictionContextCache,
+)
 import sys
 
 if sys.version_info[1] > 5:

diff --git a/tests/coverage/antlr_parser/FuncTestCaseParser.py b/tests/coverage/antlr_parser/FuncTestCaseParser.py
@@ -1,7 +1,19 @@
 # Generated from FuncTestCaseParser.g4 by ANTLR 4.13.2
 # encoding: utf-8
-from antlr4 import *
-from io import StringIO
+from antlr4 import (
+    ATNDeserializer,
+    DFA,
+    NoViableAltException,
+    ParseTreeListener,
+    ParseTreeVisitor,
+    Parser,
+    ParserATNSimulator,
+    ParserRuleContext,
+    PredictionContextCache,
+    RecognitionException,
+    Token,
+    TokenStream,
+)
 import sys
 
 if sys.version_info[1] > 5:

diff --git a/tests/coverage/antlr_parser/FuncTestCaseParserListener.py b/tests/coverage/antlr_parser/FuncTestCaseParserListener.py
@@ -1,5 +1,5 @@
 # Generated from FuncTestCaseParser.g4 by ANTLR 4.13.2
-from antlr4 import *
+from antlr4 import ParseTreeListener, ParseTreeVisitor
 
 if "." in __name__:
     from .FuncTestCaseParser import FuncTestCaseParser

diff --git a/tests/coverage/antlr_parser/FuncTestCaseParserVisitor.py b/tests/coverage/antlr_parser/FuncTestCaseParserVisitor.py
@@ -1,5 +1,5 @@
 # Generated from FuncTestCaseParser.g4 by ANTLR 4.13.2
-from antlr4 import *
+from antlr4 import ParseTreeVisitor
 
 if "." in __name__:
     from .FuncTestCaseParser import FuncTestCaseParser

diff --git a/tests/coverage/antlr_parser/SubstraitLexer.py b/tests/coverage/antlr_parser/SubstraitLexer.py
@@ -1,6 +1,11 @@
 # Generated from SubstraitLexer.g4 by ANTLR 4.13.2
-from antlr4 import *
-from io import StringIO
+from antlr4 import (
+    ATNDeserializer,
+    DFA,
+    Lexer,
+    LexerATNSimulator,
+    PredictionContextCache,
+)
 import sys
 
 if sys.version_info[1] > 5:

diff --git a/tests/coverage/test_coverage.py b/tests/coverage/test_coverage.py
@@ -25,7 +25,6 @@ def test_parse_basic_example():
 add(120::i8, 10::i8) [overflow:ERROR] = <!ERROR>
 """
 
-    test_str = header + tests
     test_file = parse_string(header + tests)
     assert len(test_file.testcases) == 3