File size: 2,786 Bytes
d9d9220 |
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 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 |
from typing import Dict, NamedTuple
class ExpectedChunk(NamedTuple):
name: str
docstring: str
content_snippet: str # A unique part of the content that should be present
# Test file contents
SIMPLE_FUCNTION_CLEANED_DOCSTRING = "Say hello to the world"
SIMPLE_FUNCTION_DOCSTRING = f'\"\"\"{SIMPLE_FUCNTION_CLEANED_DOCSTRING}\"\"\"'
SIMPLE_FUNCTION = f'''
{SIMPLE_FUNCTION_DOCSTRING}
def hello_world():
return "Hello, World!"
'''
SIMPLE_CLASS_DOCSTRING = f'\"\"\"A simple class for testing\"\"\"'
SIMPLE_CLASS = f'''
{SIMPLE_CLASS_DOCSTRING}
class SimpleClass:
def __init__(self):
self.value = 42
def get_value(self):
return self.value
'''
NESTED_OUTER_CLASS_DOCSTRING = "#Outer class docstring"
NESTED_CLASS = f'''
{NESTED_OUTER_CLASS_DOCSTRING}
class OuterClass:
class InnerClass:
"""Inner class docstring"""
def inner_method(self):
return "inner"
def outer_method(self):
return "outer"
'''
COMPLEX_FUNCTION_DOCSTRING = f"""\"\"\"
A complex function with type hints and docstring
Args:
param1: First parameter
param2: Optional second parameter
Returns:
List of strings
\"\"\""""
COMPLEX_CLASS_DOCSTRING = "# # #Complex class implementation"
COMPLEX_FILE = f'''
import os
from typing import List, Optional
{COMPLEX_FUNCTION_DOCSTRING}
def complex_function(param1: str, param2: Optional[int] = None) -> List[str]:
results = []
if param2 is not None:
results.extend([param1] * param2)
return results
# Some comment
CONSTANT = 42
{COMPLEX_CLASS_DOCSTRING}
class ComplexClass:
"""Complex class implementation Test Test"""
def __init__(self):
self._value = None
'''
INVALID_SYNTAX = '''def invalid_syntax(:'''
# Expected test results
SIMPLE_FILE_EXPECTATIONS = {
'hello_world': ExpectedChunk(
name="hello_world",
docstring=SIMPLE_FUNCTION_DOCSTRING,
content_snippet='return "Hello, World!"'
),
'SimpleClass': ExpectedChunk(
name="SimpleClass",
docstring=SIMPLE_CLASS_DOCSTRING,
content_snippet='self.value = 42'
)
}
NESTED_CLASS_EXPECTATIONS = {
'OuterClass': ExpectedChunk(
name="OuterClass",
docstring=NESTED_OUTER_CLASS_DOCSTRING,
content_snippet='class InnerClass'
)
}
COMPLEX_FILE_EXPECTATIONS = {
'complex_function': ExpectedChunk(
name="complex_function",
docstring=COMPLEX_FUNCTION_DOCSTRING,
content_snippet='List[str]'
),
'ComplexClass': ExpectedChunk(
name="ComplexClass",
docstring=COMPLEX_CLASS_DOCSTRING,
content_snippet='_value = None'
)
}
TEST_FILES = {
'simple.py': SIMPLE_FUNCTION + SIMPLE_CLASS,
'nested.py': NESTED_CLASS,
'complex.py': COMPLEX_FILE
}
|