def test_cell_values_finder_with_neighbors(self):
cell_values_neighbors_results = [
([CellValue(3143), CellValue(2525)], [
NeighborCell(df=self.duplicates_df,
cell_value=CellValue(4690),
cell_offset=CellOffset(col=-1, row=0))
], [CellPosition(col=2, row=5)]),
]
finder = AllCellPositionsFinder(df=self.duplicates_df)
for cell_values, neighbors, results in cell_values_neighbors_results:
finder.value_finder = ExactValuesFinder(cell_values=cell_values)
finder.neighbors_container = NeighborsContainer(
neighbors=neighbors)
finder_results = finder.get_all_positions()
self._check_results(expected_result=results,
finder_result=finder_results)
def test_cell_value_finder_with_neighbors(self):
cell_values_neighbors_results = [
(CellValue(2525), [
NeighborCell(df=self.duplicates_df,
cell_value=CellValue('Общий итог'),
cell_offset=CellOffset(col=-2, row=6))
], [CellPosition(col=2, row=5)]),
(CellValue(0), [
NeighborCell(df=self.duplicates_df,
cell_value=CellValue(),
cell_offset=CellOffset(col=0, row=-1))
], [CellPosition(col=5, row=6)]),
(CellValue('Общий итог'), [
NeighborCell(df=self.duplicates_df,
cell_value=CellValue(),
cell_offset=CellOffset(col=0, row=1))
], [CellPosition(col=6, row=0),
CellPosition(col=0, row=11)]),
(CellValue('Общий итог'), [
NeighborCell(df=self.duplicates_df,
cell_value=CellValue(''),
cell_offset=CellOffset(col=0, row=1))
], [CellPosition(col=6, row=0),
CellPosition(col=0, row=11)]),
(CellValue('Общий итог'), [
NeighborCell(df=self.duplicates_df,
cell_value=CellValue(),
cell_offset=CellOffset(col=0, row=1)),
NeighborCell(df=self.duplicates_df,
cell_value=CellValue(),
cell_offset=CellOffset(col=1, row=1)),
], [CellPosition(col=0, row=11)]),
(CellValue('Общий итог'), [
NeighborCell(df=self.duplicates_df,
cell_value=CellValue(),
cell_offset=CellOffset(col=0, row=1)),
NeighborCell(df=self.duplicates_df,
cell_value=CellValue(''),
cell_offset=CellOffset(col=1, row=1)),
], [CellPosition(col=0, row=11)]),
(CellValue('Общий итог'), [
NeighborCell(df=self.duplicates_df,
cell_value=CellValue(''),
cell_offset=CellOffset(col=0, row=1)),
NeighborCell(df=self.duplicates_df,
cell_value=CellValue(''),
cell_offset=CellOffset(col=1, row=1)),
], [CellPosition(col=0, row=11)]),
(CellValue('Общий итог'), [
NeighborCell(df=self.duplicates_df,
cell_value=CellValue(''),
cell_offset=CellOffset(col=0, row=1)),
NeighborCell(df=self.duplicates_df,
cell_value=CellValue(),
cell_offset=CellOffset(col=1, row=1)),
], [CellPosition(col=0, row=11)]),
]
finder = AllCellPositionsFinder(df=self.duplicates_df)
for cell_value, neighbors, results in cell_values_neighbors_results:
finder.value_finder = ExactValueFinder(cell_value=cell_value)
finder.neighbors_container = NeighborsContainer(
neighbors=neighbors)
finder_results = finder.get_all_positions()
self._check_results(expected_result=results,
finder_result=finder_results)