mirror of
https://github.com/codeflash-ai/codeflash.git
synced 2026-05-04 18:25:17 +00:00
test: update expected markdown ordering to match target-file-first change
This commit is contained in:
Kevin Turcios
parent
6ad7ea49f6
commit
6330005999
1 changed files with 94 additions and 94 deletions
|
|
@ -243,6 +243,14 @@ def test_bubble_sort_helper() -> None:
|
|||
hashing_context = code_ctx.hashing_code_context
|
||||
|
||||
expected_read_write_context = """
|
||||
```python:code_to_optimize/code_directories/retriever/bubble_sort_imported.py
|
||||
from bubble_sort_with_math import sorter
|
||||
|
||||
|
||||
def sort_from_another_file(arr):
|
||||
sorted_arr = sorter(arr)
|
||||
return sorted_arr
|
||||
```
|
||||
```python:code_to_optimize/code_directories/retriever/bubble_sort_with_math.py
|
||||
import math
|
||||
|
||||
|
|
@ -253,14 +261,6 @@ def sorter(arr):
|
|||
print(x)
|
||||
return arr
|
||||
```
|
||||
```python:code_to_optimize/code_directories/retriever/bubble_sort_imported.py
|
||||
from bubble_sort_with_math import sorter
|
||||
|
||||
|
||||
def sort_from_another_file(arr):
|
||||
sorted_arr = sorter(arr)
|
||||
return sorted_arr
|
||||
```
|
||||
"""
|
||||
expected_read_only_context = ""
|
||||
|
||||
|
|
@ -1178,6 +1178,26 @@ def test_repo_helper() -> None:
|
|||
hashing_context = code_ctx.hashing_code_context
|
||||
path_to_globals = project_root / "globals.py"
|
||||
expected_read_write_context = f"""
|
||||
```python:{path_to_file.relative_to(project_root)}
|
||||
import requests
|
||||
from globals import API_URL
|
||||
from utils import DataProcessor
|
||||
|
||||
|
||||
def fetch_and_process_data():
|
||||
# Use the global variable for the request
|
||||
response = requests.get(API_URL)
|
||||
response.raise_for_status()
|
||||
|
||||
raw_data = response.text
|
||||
|
||||
# Use code from another file (utils.py)
|
||||
processor = DataProcessor()
|
||||
processed = processor.process_data(raw_data)
|
||||
processed = processor.add_prefix(processed)
|
||||
|
||||
return processed
|
||||
```
|
||||
```python:{path_to_globals.relative_to(project_root)}
|
||||
# Define a global variable
|
||||
API_URL = "https://api.example.com/data"
|
||||
|
|
@ -1201,26 +1221,6 @@ class DataProcessor:
|
|||
\"\"\"Add a prefix to the processed data.\"\"\"
|
||||
return prefix + data
|
||||
```
|
||||
```python:{path_to_file.relative_to(project_root)}
|
||||
import requests
|
||||
from globals import API_URL
|
||||
from utils import DataProcessor
|
||||
|
||||
|
||||
def fetch_and_process_data():
|
||||
# Use the global variable for the request
|
||||
response = requests.get(API_URL)
|
||||
response.raise_for_status()
|
||||
|
||||
raw_data = response.text
|
||||
|
||||
# Use code from another file (utils.py)
|
||||
processor = DataProcessor()
|
||||
processed = processor.process_data(raw_data)
|
||||
processed = processor.add_prefix(processed)
|
||||
|
||||
return processed
|
||||
```
|
||||
"""
|
||||
expected_read_only_context = f"""
|
||||
```python:{path_to_utils.relative_to(project_root)}
|
||||
|
|
@ -1278,6 +1278,25 @@ def test_repo_helper_of_helper() -> None:
|
|||
hashing_context = code_ctx.hashing_code_context
|
||||
path_to_globals = project_root / "globals.py"
|
||||
expected_read_write_context = f"""
|
||||
```python:{path_to_file.relative_to(project_root)}
|
||||
import requests
|
||||
from globals import API_URL
|
||||
from utils import DataProcessor
|
||||
|
||||
|
||||
def fetch_and_transform_data():
|
||||
# Use the global variable for the request
|
||||
response = requests.get(API_URL)
|
||||
|
||||
raw_data = response.text
|
||||
|
||||
# Use code from another file (utils.py)
|
||||
processor = DataProcessor()
|
||||
processed = processor.process_data(raw_data)
|
||||
transformed = processor.transform_data(processed)
|
||||
|
||||
return transformed
|
||||
```
|
||||
```python:{path_to_globals.relative_to(project_root)}
|
||||
# Define a global variable
|
||||
API_URL = "https://api.example.com/data"
|
||||
|
|
@ -1302,25 +1321,6 @@ class DataProcessor:
|
|||
\"\"\"Transform the processed data\"\"\"
|
||||
return DataTransformer().transform(data)
|
||||
```
|
||||
```python:{path_to_file.relative_to(project_root)}
|
||||
import requests
|
||||
from globals import API_URL
|
||||
from utils import DataProcessor
|
||||
|
||||
|
||||
def fetch_and_transform_data():
|
||||
# Use the global variable for the request
|
||||
response = requests.get(API_URL)
|
||||
|
||||
raw_data = response.text
|
||||
|
||||
# Use code from another file (utils.py)
|
||||
processor = DataProcessor()
|
||||
processed = processor.process_data(raw_data)
|
||||
transformed = processor.transform_data(processed)
|
||||
|
||||
return transformed
|
||||
```
|
||||
"""
|
||||
expected_read_only_context = f"""
|
||||
```python:{path_to_utils.relative_to(project_root)}
|
||||
|
|
@ -1384,14 +1384,6 @@ def test_repo_helper_of_helper_same_class() -> None:
|
|||
read_write_context, read_only_context = code_ctx.read_writable_code, code_ctx.read_only_context_code
|
||||
hashing_context = code_ctx.hashing_code_context
|
||||
expected_read_write_context = f"""
|
||||
```python:{path_to_transform_utils.relative_to(project_root)}
|
||||
class DataTransformer:
|
||||
def __init__(self):
|
||||
self.data = None
|
||||
|
||||
def transform_using_own_method(self, data):
|
||||
return self.transform(data)
|
||||
```
|
||||
```python:{path_to_utils.relative_to(project_root)}
|
||||
import math
|
||||
from transform_utils import DataTransformer
|
||||
|
|
@ -1408,6 +1400,14 @@ class DataProcessor:
|
|||
\"\"\"Transform the processed data using own method\"\"\"
|
||||
return DataTransformer().transform_using_own_method(data)
|
||||
```
|
||||
```python:{path_to_transform_utils.relative_to(project_root)}
|
||||
class DataTransformer:
|
||||
def __init__(self):
|
||||
self.data = None
|
||||
|
||||
def transform_using_own_method(self, data):
|
||||
return self.transform(data)
|
||||
```
|
||||
"""
|
||||
expected_read_only_context = f"""
|
||||
```python:{path_to_transform_utils.relative_to(project_root)}
|
||||
|
|
@ -1465,14 +1465,6 @@ def test_repo_helper_of_helper_same_file() -> None:
|
|||
read_write_context, read_only_context = code_ctx.read_writable_code, code_ctx.read_only_context_code
|
||||
hashing_context = code_ctx.hashing_code_context
|
||||
expected_read_write_context = f"""
|
||||
```python:{path_to_transform_utils.relative_to(project_root)}
|
||||
class DataTransformer:
|
||||
def __init__(self):
|
||||
self.data = None
|
||||
|
||||
def transform_using_same_file_function(self, data):
|
||||
return update_data(data)
|
||||
```
|
||||
```python:{path_to_utils.relative_to(project_root)}
|
||||
import math
|
||||
from transform_utils import DataTransformer
|
||||
|
|
@ -1489,6 +1481,14 @@ class DataProcessor:
|
|||
\"\"\"Transform the processed data using a function from the same file\"\"\"
|
||||
return DataTransformer().transform_using_same_file_function(data)
|
||||
```
|
||||
```python:{path_to_transform_utils.relative_to(project_root)}
|
||||
class DataTransformer:
|
||||
def __init__(self):
|
||||
self.data = None
|
||||
|
||||
def transform_using_same_file_function(self, data):
|
||||
return update_data(data)
|
||||
```
|
||||
"""
|
||||
expected_read_only_context = f"""
|
||||
```python:{path_to_transform_utils.relative_to(project_root)}
|
||||
|
|
@ -1605,6 +1605,17 @@ def test_repo_helper_circular_dependency() -> None:
|
|||
read_write_context, read_only_context = code_ctx.read_writable_code, code_ctx.read_only_context_code
|
||||
hashing_context = code_ctx.hashing_code_context
|
||||
expected_read_write_context = f"""
|
||||
```python:{path_to_transform_utils.relative_to(project_root)}
|
||||
from code_to_optimize.code_directories.retriever.utils import DataProcessor
|
||||
|
||||
|
||||
class DataTransformer:
|
||||
def __init__(self):
|
||||
self.data = None
|
||||
|
||||
def circular_dependency(self, data):
|
||||
return DataProcessor().circular_dependency(data)
|
||||
```
|
||||
```python:{path_to_utils.relative_to(project_root)}
|
||||
import math
|
||||
from transform_utils import DataTransformer
|
||||
|
|
@ -1621,17 +1632,6 @@ class DataProcessor:
|
|||
\"\"\"Test circular dependency\"\"\"
|
||||
return DataTransformer().circular_dependency(data)
|
||||
```
|
||||
```python:{path_to_transform_utils.relative_to(project_root)}
|
||||
from code_to_optimize.code_directories.retriever.utils import DataProcessor
|
||||
|
||||
|
||||
class DataTransformer:
|
||||
def __init__(self):
|
||||
self.data = None
|
||||
|
||||
def circular_dependency(self, data):
|
||||
return DataProcessor().circular_dependency(data)
|
||||
```
|
||||
"""
|
||||
expected_read_only_context = f"""
|
||||
```python:{path_to_utils.relative_to(project_root)}
|
||||
|
|
@ -1796,6 +1796,12 @@ def function_to_optimize():
|
|||
```
|
||||
"""
|
||||
expected_read_write_context = f"""
|
||||
```python:{path_to_fto.relative_to(project_root)}
|
||||
import code_to_optimize.code_directories.retriever.main
|
||||
|
||||
def function_to_optimize():
|
||||
return code_to_optimize.code_directories.retriever.main.fetch_and_transform_data()
|
||||
```
|
||||
```python:{path_to_main.relative_to(project_root)}
|
||||
import requests
|
||||
from globals import API_URL
|
||||
|
|
@ -1815,12 +1821,6 @@ def fetch_and_transform_data():
|
|||
|
||||
return transformed
|
||||
```
|
||||
```python:{path_to_fto.relative_to(project_root)}
|
||||
import code_to_optimize.code_directories.retriever.main
|
||||
|
||||
def function_to_optimize():
|
||||
return code_to_optimize.code_directories.retriever.main.fetch_and_transform_data()
|
||||
```
|
||||
"""
|
||||
assert read_write_context.markdown.strip() == expected_read_write_context.strip()
|
||||
assert read_only_context.strip() == expected_read_only_context.strip()
|
||||
|
|
@ -2244,6 +2244,20 @@ def get_system_details():
|
|||
# The expected contexts
|
||||
relative_path = file_path.relative_to(project_root)
|
||||
expected_read_write_context = f"""
|
||||
```python:{main_file_path.resolve().relative_to(opt.args.project_root.resolve())}
|
||||
import utility_module
|
||||
|
||||
class Calculator:
|
||||
def __init__(self, precision="high", fallback_precision=None, mode="standard"):
|
||||
# This is where we use the imported module
|
||||
self.precision = utility_module.select_precision(precision, fallback_precision)
|
||||
self.mode = mode
|
||||
|
||||
# Using variables from the utility module
|
||||
self.backend = utility_module.CALCULATION_BACKEND
|
||||
self.system = utility_module.SYSTEM_TYPE
|
||||
self.default_precision = utility_module.DEFAULT_PRECISION
|
||||
```
|
||||
```python:utility_module.py
|
||||
import sys
|
||||
|
||||
|
|
@ -2291,20 +2305,6 @@ def select_precision(precision, fallback_precision):
|
|||
else:
|
||||
return DEFAULT_PRECISION
|
||||
```
|
||||
```python:{main_file_path.resolve().relative_to(opt.args.project_root.resolve())}
|
||||
import utility_module
|
||||
|
||||
class Calculator:
|
||||
def __init__(self, precision="high", fallback_precision=None, mode="standard"):
|
||||
# This is where we use the imported module
|
||||
self.precision = utility_module.select_precision(precision, fallback_precision)
|
||||
self.mode = mode
|
||||
|
||||
# Using variables from the utility module
|
||||
self.backend = utility_module.CALCULATION_BACKEND
|
||||
self.system = utility_module.SYSTEM_TYPE
|
||||
self.default_precision = utility_module.DEFAULT_PRECISION
|
||||
```
|
||||
"""
|
||||
expected_read_only_context = """
|
||||
```python:utility_module.py
|
||||
|
|
|
|||
Loading…
Reference in a new issue