def get_column_id(operation: Operation, column_name: str) -> int: column_found, column_id = operation.column_name_to_id(column_name) if not column_found: print('Column ' + column_name + ' not found.', file=sys.stderr) sys.exit(-1) return column_id
def test_column_name_to_id(dataset, mock_dataset_column_name_to_id) -> None: mock_dataset_column_name_to_id.return_value = True, 1 operation = Operation(dataset) assert (True, 1) == operation.column_name_to_id('something')