| 3 | |
| 4 | |
| 5 | def fix(data): |
| 6 | # fix 53 https://github.com/evalplus/evalplus/issues/8 |
| 7 | check_id(data, 53) |
| 8 | data[53]["contract"] = ( |
| 9 | '\n assert isinstance(x, int), "invalid inputs" # $_CONTRACT_$' |
| 10 | + '\n assert isinstance(y, int), "invalid inputs" # $_CONTRACT_$\n' |
| 11 | ) |
| 12 | data[53]["plus_input"] = [ |
| 13 | x |
| 14 | for x in data[53]["plus_input"] |
| 15 | if isinstance(x[0], int) and isinstance(x[1], int) |
| 16 | ] |
| 17 | |
| 18 | # fix 0 |
| 19 | check_id(data, 0) |
| 20 | data[0]["contract"] = ( |
| 21 | '\n assert isinstance(threshold, float) and threshold > 0, "invalid inputs" # $_CONTRACT_$' |
| 22 | + '\n assert isinstance(numbers, list), "invalid inputs" # $_CONTRACT_$' |
| 23 | + '\n assert all([isinstance(v, (int, float)) for v in numbers]), "invalid inputs" # $_CONTRACT_$\n' |
| 24 | ) |
| 25 | data[0]["plus_input"] = [ |
| 26 | x |
| 27 | for x in data[0]["plus_input"] |
| 28 | if isinstance(x[1], float) and x[1] > 0 and isinstance(x[0], list) |
| 29 | ] |
| 30 | |
| 31 | # fix 3 |
| 32 | check_id(data, 3) |
| 33 | data[3]["contract"] = ( |
| 34 | '\n assert type(operations) == list, "invalid inputs" # $_CONTRACT_$' |
| 35 | + '\n assert all([isinstance(v, int) for v in operations]), "invalid inputs" # $_CONTRACT_$\n' |
| 36 | ) |
| 37 | data[3]["plus_input"] = [x for x in data[3]["plus_input"] if isinstance(x[0], list)] |
| 38 | |
| 39 | # fix 9 |
| 40 | check_id(data, 9) |
| 41 | data[9]["contract"] = ( |
| 42 | '\n assert isinstance(numbers, list), "invalid inputs" # $_CONTRACT_$' |
| 43 | + '\n assert all([isinstance(v, int) for v in numbers]), "invalid inputs" # $_CONTRACT_$\n' |
| 44 | ) |
| 45 | data[9]["plus_input"] = [x for x in data[9]["plus_input"] if isinstance(x[0], list)] |
| 46 | |
| 47 | # fix 148 |
| 48 | check_id(data, 148) |
| 49 | data[148][ |
| 50 | "contract" |
| 51 | ] = '\n assert isinstance(planet1, str) and isinstance(planet2, str), "invalid inputs" # $_CONTRACT_$\n' |
| 52 | data[148]["plus_input"] = [ |
| 53 | x |
| 54 | for x in data[148]["plus_input"] |
| 55 | if isinstance(x[0], str) and isinstance(x[1], str) |
| 56 | ] |
| 57 | |
| 58 | # minor format fix 75 |
| 59 | check_id(data, 75) |
| 60 | data[75]["contract"] = ( |
| 61 | '\n assert type(a) == int, "invalid inputs" # $_CONTRACT_$' |
| 62 | + '\n assert a < 100, "invalid inputs" # $_CONTRACT_$\n' |