gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Comment out profiling

Browse source
This commit is contained in:
vasilije 2025-01-05 20:55:47 +01:00
parent 325102562a
commit 0983e95a33
1 changed files with 17 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
cognee/tests/unit/processing/utils/utils_test.py
View file

@ -35,24 +35,24 @@ def test_get_anonymous_id(mock_open_file, mock_makedirs, temp_dir):
assert len(anon_id) > 0
@patch("requests.post")
def test_send_telemetry(mock_post):
mock_post.return_value.status_code = 200
send_telemetry("test_event", "test_user", {"key": "value"})
mock_post.assert_called_once()
args, kwargs = mock_post.call_args
assert kwargs["json"]["event_name"] == "test_event"
@patch("tiktoken.encoding_for_model")
def test_num_tokens_from_string(mock_encoding):
mock_encoding.return_value.encode = lambda x: list(x)
assert num_tokens_from_string("hello", "test_encoding") == 5
assert num_tokens_from_string("world", "test_encoding") == 5
# @patch("requests.post")
# def test_send_telemetry(mock_post):
# mock_post.return_value.status_code = 200
#
# send_telemetry("test_event", "test_user", {"key": "value"})
# mock_post.assert_called_once()
#
# args, kwargs = mock_post.call_args
# assert kwargs["json"]["event_name"] == "test_event"
#
# @patch("tiktoken.encoding_for_model")
# def test_num_tokens_from_string(mock_encoding):
# mock_encoding.return_value.encode = lambda x: list(x)
#
# assert num_tokens_from_string("hello", "test_encoding") == 5
# assert num_tokens_from_string("world", "test_encoding") == 5
#
@patch("builtins.open", new_callable=mock_open, read_data=b"test_data")
def test_get_file_content_hash_file(mock_open_file):