def main() -> None: args = parse_args(sys.argv[1:]) with PerformanceChecker('Data loading'): dataset = load_data(args.file) operation = Operation(dataset) query = create_query(operation, args) with PerformanceChecker('Operation'): results = operation.execute(query) print('Results: ', results)
def check_results(operation: Operation, query: Query, expected_results: Dict) -> None: result = operation.execute(query) assert result == expected_results