Skip to content

Commit 50858d8

Browse files
committed
Removes mock package dependency
1 parent eb8d44b commit 50858d8

File tree

5 files changed

+4
-7
lines changed

5 files changed

+4
-7
lines changed

tests/cross_agent/test_docker_container_id.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515
import json
1616
import os
1717

18-
import mock
1918
import pytest
2019

2120
import newrelic.common.utilization as u
@@ -43,7 +42,7 @@ def _mock_open(filename, mode):
4342
raise FileNotFoundError
4443
elif filename == "/proc/self/cgroup":
4544
return mock_file
46-
raise RuntimeError
45+
raise RuntimeError()
4746

4847
return _mock_open
4948

tests/cross_agent/test_docker_container_id_v2.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515
import json
1616
import os
1717

18-
import mock
1918
import pytest
2019

2120
import newrelic.common.utilization as u
@@ -43,7 +42,7 @@ def _mock_open(filename, mode):
4342
raise FileNotFoundError
4443
elif filename == "/proc/self/mountinfo":
4544
return mock_file
46-
raise RuntimeError
45+
raise RuntimeError()
4746

4847
return _mock_open
4948

tests/mlmodel_langchain/test_chain.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from langchain.chains.combine_documents import create_stuff_documents_chain
2323
from langchain.chains.openai_functions import create_structured_output_chain, create_structured_output_runnable
2424
from langchain_community.vectorstores.faiss import FAISS
25-
from mock import patch
25+
from unittest.mock import patch
2626
from testing_support.fixtures import reset_core_stats_engine, validate_attributes
2727
from testing_support.ml_testing_utils import (
2828
disabled_ai_monitoring_record_content_settings,

tests/mlmodel_langchain/test_tool.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
import pydantic_core
2121
import pytest
2222
from langchain.tools import tool
23-
from mock import patch
23+
from unittest.mock import patch
2424
from testing_support.fixtures import reset_core_stats_engine, validate_attributes
2525
from testing_support.ml_testing_utils import (
2626
disabled_ai_monitoring_record_content_settings,

tox.ini

-1
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,6 @@ deps =
250250
component_tastypie-tastypielatest: django<4.1
251251
component_tastypie-tastypielatest: asgiref<3.7.1 # asgiref==3.7.1 only suppport Python 3.10+
252252
coroutines_asyncio-{py37,py38,py39,py310,py311,py312,py313}: uvloop
253-
cross_agent: mock==1.0.1
254253
cross_agent: requests
255254
datastore_asyncpg: asyncpg
256255
datastore_aiomcache: aiomcache

0 commit comments

Comments
 (0)