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')
