Merge branch 'main' of github.com:alphanome-ai/sec-parser

tests/unit/processing_engine/test_html_tag.py

@@ -1,6 +1,3 @@
-from unittest import mock
-from unittest.mock import Mock
-
 import bs4
 import pytest
 from bs4 import NavigableString

tests/unit/processing_steps/individual_semantic_element_extractor/single_element_checks/test_top_level_section_title_check.py

@@ -1,5 +1,3 @@
-from unittest.mock import Mock
-
 import bs4
 import pytest
 

tests/unit/processing_steps/test_empty_element_classifier.py

@@ -1,5 +1,3 @@
-from unittest.mock import Mock
-
 import pytest
 
 from sec_parser.processing_engine.core import Edgar10QParser

tests/unit/processing_steps/test_supplementary_text_classifier.py

@@ -1,5 +1,3 @@
-from unittest.mock import Mock
-
 import bs4
 import pytest
 

tests/unit/semantic_elements/composite_semantic_element/test_to_dict.py

@@ -1,5 +1,3 @@
-from unittest.mock import Mock
-
 import bs4
 
 from sec_parser.processing_engine.html_tag import HtmlTag

tests/unit/semantic_elements/table_element/test_table_element.py

@@ -38,13 +38,13 @@ def test_table_element_get_summary(test_case):
     numbers = test_case["numbers"]
     text = test_case["text"]
     expected_summary = test_case["expected_summary"]
-    mock_semantic_element = Mock()
-    mock_semantic_element.text = text
-    mock_semantic_element.get_approx_table_metrics.return_value = ApproxTableMetrics(
+    mock_html_tag = Mock()
+    mock_html_tag.text = text
+    mock_html_tag.get_approx_table_metrics.return_value = ApproxTableMetrics(
         rows,
         numbers,
     )
-    table_element = TableElement(mock_semantic_element)
+    table_element = TableElement(mock_html_tag)
 
     # Act
     result = table_element.get_summary()
@@ -55,15 +55,15 @@ def test_table_element_get_summary(test_case):
 
 def test_to_dict():
     # Arrange
-    mock_semantic_element = Mock(spec=HtmlTag)
-    mock_semantic_element.get_approx_table_metrics.return_value = ApproxTableMetrics(
+    mock_html_tag = Mock(spec=HtmlTag)
+    mock_html_tag.get_approx_table_metrics.return_value = ApproxTableMetrics(
         5, 6
     )
-    mock_semantic_element.to_dict.return_value = (
+    mock_html_tag.to_dict.return_value = (
         {}
     )  # Mock the to_dict method to return an empty dictionary
 
-    table_element = TableElement(mock_semantic_element)
+    table_element = TableElement(mock_html_tag)
 
     # Act
     actual = table_element.to_dict(include_previews=True)