|
|
|
|
@ -342,3 +342,26 @@ def test_extract_text_from_excel_all_sheets_fail(mock_excel_file):
|
|
|
|
|
assert result == ""
|
|
|
|
|
|
|
|
|
|
assert mock_excel_instance.parse.call_count == 2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@patch("pandas.ExcelFile")
|
|
|
|
|
def test_extract_text_from_excel_numeric_type_column(mock_excel_file):
|
|
|
|
|
"""Test extracting text from Excel file with numeric column names."""
|
|
|
|
|
|
|
|
|
|
# Test numeric type column
|
|
|
|
|
data = {1: ["Test"], 1.1: ["Test"]}
|
|
|
|
|
|
|
|
|
|
df = pd.DataFrame(data)
|
|
|
|
|
|
|
|
|
|
# Mock ExcelFile
|
|
|
|
|
mock_excel_instance = Mock()
|
|
|
|
|
mock_excel_instance.sheet_names = ["Sheet1"]
|
|
|
|
|
mock_excel_instance.parse.return_value = df
|
|
|
|
|
mock_excel_file.return_value = mock_excel_instance
|
|
|
|
|
|
|
|
|
|
file_content = b"fake_excel_content"
|
|
|
|
|
result = _extract_text_from_excel(file_content)
|
|
|
|
|
|
|
|
|
|
expected_manual = "| 1.0 | 1.1 |\n| --- | --- |\n| Test | Test |\n\n"
|
|
|
|
|
|
|
|
|
|
assert expected_manual == result
|
|
|
|
|
|