diff --git a/cellveyor/data.py b/cellveyor/data.py index 12090cf..f61dcdd 100644 --- a/cellveyor/data.py +++ b/cellveyor/data.py @@ -16,7 +16,6 @@ def access_dataframes(spreadsheet_file: Path) -> Dict[str, pandas.DataFrame]: name_to_dataframe_dict = pandas.read_excel(spreadsheet_file, sheet_name=None) return name_to_dataframe_dict - def key_attribute_column_filter( sheet_dataframe: pandas.DataFrame, key_attribute_name: str, diff --git a/spreadsheets/example_spreadsheet.xlsx b/spreadsheets/example_spreadsheet.xlsx new file mode 100644 index 0000000..e933f72 Binary files /dev/null and b/spreadsheets/example_spreadsheet.xlsx differ diff --git a/tests/test_data.py b/tests/test_data.py index 823cb5b..097feb3 100644 --- a/tests/test_data.py +++ b/tests/test_data.py @@ -8,10 +8,8 @@ def test_access_dataframes() -> None: - spreadsheet_file = "spreadsheets/fake_spreadsheet.xlsx" - result = data.access_dataframes(spreadsheet_file) - expected_sheet1 = pd.DataFrame({'column1': [1, 2, 3], 'column2': ['A', 'B', 'C']}) - expected_sheet2 = pd.DataFrame({'columnA': [4, 5, 6], 'columnB': ['X', 'Y', 'Z']}) - - assert pd.DataFrame.equals(result['Sheet1'], expected_sheet1) - assert pd.DataFrame.equals(result['Sheet2'], expected_sheet2) \ No newline at end of file + spreadsheet_file_one = "spreadsheets/fake_spreadsheet.xlsx" + spreadsheet_file_two = "spreadsheets/example_spreadsheet.xlsx" + test_one= data.access_dataframes(spreadsheet_file_one) + test_two = data.access_dataframes(spreadsheet_file_two) + assert test_one != test_two