FilesExpand file tree

 main

/

overwrite_expected_slice.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

90 lines (74 loc) · 3 KB

 main

/

overwrite_expected_slice.py

Copy path

Top

File metadata and controls

• Code
• Blame

90 lines (74 loc) · 3 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

# coding=utf-8

# Copyright 2025 The HuggingFace Inc. team.

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

# http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

import argparse

from collections import defaultdict

def overwrite_file(file, class_name, test_name, correct_line, done_test):

_id = f"{file}_{class_name}_{test_name}"

done_test[_id] += 1

with open(file, "r") as f:

lines = f.readlines()

class_regex = f"class {class_name}("

test_regex = f"{4 * ' '}def {test_name}("

line_begin_regex = f"{8 * ' '}{correct_line.split()[0]}"

another_line_begin_regex = f"{16 * ' '}{correct_line.split()[0]}"

in_class = False

in_func = False

in_line = False

insert_line = False

count = 0

spaces = 0

new_lines = []

for line in lines:

if line.startswith(class_regex):

in_class = True

elif in_class and line.startswith(test_regex):

in_func = True

elif in_class and in_func and (line.startswith(line_begin_regex) or line.startswith(another_line_begin_regex)):

spaces = len(line.split(correct_line.split()[0])[0])

count += 1

if count == done_test[_id]:

in_line = True

if in_class and in_func and in_line:

if ")" not in line:

continue

else:

insert_line = True

if in_class and in_func and in_line and insert_line:

new_lines.append(f"{spaces * ' '}{correct_line}")

in_class = in_func = in_line = insert_line = False

else:

new_lines.append(line)

with open(file, "w") as f:

for line in new_lines:

f.write(line)

def main(correct, fail=None):

if fail is not None:

with open(fail, "r") as f:

test_failures = {l.strip() for l in f.readlines()}

else:

test_failures = None

with open(correct, "r") as f:

correct_lines = f.readlines()

done_tests = defaultdict(int)

for line in correct_lines:

file, class_name, test_name, correct_line = line.split("::")

if test_failures is None or "::".join([file, class_name, test_name]) in test_failures:

overwrite_file(file, class_name, test_name, correct_line, done_tests)

if __name__ == "__main__":

parser = argparse.ArgumentParser()

parser.add_argument("--correct_filename", help="filename of tests with expected result")

parser.add_argument("--fail_filename", help="filename of test failures", type=str, default=None)

args = parser.parse_args()

main(args.correct_filename, args.fail_filename)