test: improve assertions and add new tests

Author: SunBK201

tests/test_file.py

@@ -12,7 +12,9 @@ def setUp(self):
         self.project = scubatrace.Project.create(
             str(self.project_path), language=scubatrace.language.C
         )
-        self.file = self.project.files.get("main.c")
+        file = self.project.files.get("main.c")
+        assert file is not None
+        self.file = file
 
     def test_file_create(self):
         file = scubatrace.File.create(
@@ -22,35 +24,30 @@ def test_file_create(self):
         self.assertIsNotNone(file)
 
     def test_file_imports(self):
-        assert self.file is not None
         imports = self.file.imports
         self.assertGreater(len(imports), 0)
         for imp in imports:
             self.assertTrue(imp.name in ["stdio.h", "sub.h"])
 
     def test_file_functions(self):
-        assert self.file is not None
         functions = self.file.functions
         self.assertGreater(len(functions), 0)
         for func in functions:
             self.assertIsNotNone(func.name)
 
     def test_file_function_by_line(self):
-        assert self.file is not None
         function = self.file.function_by_line(5)
         self.assertIsNotNone(function)
         assert function is not None
         self.assertEqual(function.name, "add")
 
     def test_file_statements(self):
-        assert self.file is not None
         statements = self.file.statements
         self.assertGreater(len(statements), 0)
         for stmt in statements:
             self.assertIsNotNone(stmt.text)
 
     def test_file_statement_by_line(self):
-        assert self.file is not None
         statements = self.file.statements_by_line(14)
         self.assertGreater(len(statements), 0)
         statement = statements[0]
@@ -61,14 +58,12 @@ def test_file_statement_by_line(self):
         self.assertGreater(len(self.file.statements_by_line(1)), 0)
 
     def test_file_identifiers(self):
-        assert self.file is not None
         identifiers = self.file.identifiers
         self.assertGreater(len(identifiers), 0)
         for identifier in identifiers:
             self.assertIsNotNone(identifier.name)
 
     def test_file_variables(self):
-        assert self.file is not None
         variables = self.file.variables
         self.assertGreater(len(variables), 0)
         for var in variables:
@@ -82,7 +77,6 @@ def test_file_cfg(self):
         self.assertGreater(len(cfg.edges), 0)
 
     def test_file_query(self):
-        assert self.file is not None
         query_str = """
             (call_expression
                 function: (identifier)@func

tests/test_function.py

@@ -14,9 +14,17 @@ def setUp(self):
         )
         self.file = self.project.files.get("main.c")
         assert self.file is not None
-        self.function = self.file.function_by_line(11)
+        function = self.file.function_by_line(11)
+        assert function is not None
+        self.function = function
 
     def test_function_create(self):
-        assert self.function is not None
         function = scubatrace.Function.create(self.function.node, self.function.parent)
         self.assertIsNotNone(function)
+
+    def test_function_lines(self):
+        self.assertGreater(len(self.function.lines), 0)
+
+    def test_function_parameter_lines(self):
+        self.assertEqual(len(self.function.parameter_lines), 1)
+        self.assertEqual(self.function.parameter_lines[0], 9)

tests/test_statement.py

@@ -21,7 +21,6 @@ def setUp(self):
         self.statement = statement[0]
 
     def test_statement_create(self):
-        assert self.statement is not None
         statement = scubatrace.SimpleStatement.create(
             self.statement.node, self.statement.parent
         )