File tree Expand file tree Collapse file tree 1 file changed +0
-17
lines changed Expand file tree Collapse file tree 1 file changed +0
-17
lines changed Original file line number Diff line number Diff line change 55from unittest .mock import MagicMock , patch
66
77import pytest
8- import tiktoken
98
109from codeflash .code_utils .code_utils import (
1110 cleanup_paths ,
2322from codeflash .code_utils .concolic_utils import clean_concolic_tests
2423from codeflash .code_utils .coverage_utils import generate_candidates , prepare_coverage_files
2524
26- def test_encode_str ():
27- #not testing anything, just analyzing the behavior of encoding
28- #print("\n")
29- codebases_to_try = Path (Path (__file__ ).parent .resolve () / "../code_to_optimize/" ).glob ("**/*.py" )
30- ave_ratio = []
31- max_ratio_dict = dict ()
32- for code_fn in codebases_to_try :
33- code_str = code_fn .read_text (encoding = "utf-8" )
34- if not len (code_str ) or "__init__.py" in str (code_fn ):
35- continue
36- tokenizer = tiktoken .encoding_for_model ("gpt-4o" )
37- tkt_encoded_str = tokenizer .encode (code_str )
38- code_len = len (code_str )
39- ave_ratio .append (len (tkt_encoded_str )/ code_len )
40- max_ratio_dict [len (tkt_encoded_str )/ code_len ] = code_fn
41- print (sum (ave_ratio )/ len (ave_ratio ), min (ave_ratio ), max (ave_ratio ))
4225
4326@pytest .fixture
4427def multiple_existing_and_non_existing_files (tmp_path : Path ) -> list [Path ]:
You can’t perform that action at this time.
0 commit comments