Upload test_case_gen.py with huggingface_hub

test_case_gen.py

@@ -0,0 +1,105 @@
+# from hypothesis import given
+# from hypothesis.strategies import text
+
+# @given(text())
+# def test_uppercase_conversion(s):
+    # assert not (s is None)
+    
+    
+# import ast
+
+# def extract_input_types(code_str):
+    # tree = ast.parse(code_str)
+    # input_types = []
+
+    # class InputVisitor(ast.NodeVisitor):
+        # def visit_Call(self, node):
+            # if isinstance(node.func, ast.Name) and node.func.id == 'input':
+                # # Assumes single input call, no argument parsing
+                # input_types.append('string')
+            # self.generic_visit(node)
+
+    # visitor = InputVisitor()
+    # visitor.visit(tree)
+    
+    # return input_types
+
+# # Example usage
+# code_str = """
+# t = int(input())
+# for i in range(t):
+    # n = int(input())
+    # j = 9
+    # firstdecrease = False
+    # for i in range(n):
+        # print(j, end='')
+        # if not firstdecrease:
+            # j = (j - 1) % 10
+        # else:
+            # j = (j + 1) % 10
+        # if not firstdecrease and j == 8:
+            # firstdecrease = True
+    # print()
+# """
+
+# input_types = extract_input_types(code_str)
+# print(input_types)  # ['string', 'string']
+
+
+
+# fuzz.py
+import afl
+import sys
+import io
+
+def run_code_with_input(code_str, inputs):
+    # Redirect stdout to capture print statements
+    old_stdout = sys.stdout
+    sys.stdout = io.StringIO()
+
+    # Prepare the input
+    input_str = '\n'.join(map(str, inputs)) + '\n'
+    old_stdin = sys.stdin
+    sys.stdin = io.StringIO(input_str)
+
+    try:
+        # Execute the code
+        exec(code_str)
+        # Get the output
+        output = sys.stdout.getvalue().strip()
+    finally:
+        # Reset stdout and stdin
+        sys.stdout = old_stdout
+        sys.stdin = old_stdin
+
+    return output
+
+def fuzz():
+    afl.init()
+    code_str = """
+    import sys
+    t = int(input())
+    for _ in range(t):
+        n = int(input())
+        j = 9
+        firstdecrease = False
+        for _ in range(n):
+            print(j, end='')
+            if not firstdecrease:
+                j = (j - 1) % 10
+            else:
+                j = (j + 1) % 10
+            if not firstdecrease and j == 8:
+                firstdecrease = True
+        print()
+    """
+
+    # Fuzz the code
+    while True:
+        input_data = afl.raw_input()
+        inputs = input_data.decode().strip().split('\n')
+        inputs = list(map(int, inputs))  # Convert to integers if necessary
+        run_code_with_input(code_str, inputs)
+
+if __name__ == "__main__":
+    fuzz()