File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 1
1
import asyncio
2
2
from contextlib import contextmanager
3
3
from pathlib import Path
4
+
5
+ import asyncmock
4
6
from servicex .servicex import StreamInfoPath , StreamInfoUrl
5
7
from servicex .cache import Cache
6
8
@@ -596,7 +598,7 @@ async def test_stream_awkward_data(mocker):
596
598
mock_servicex_adaptor = MockServiceXAdaptor (mocker , "123-456" )
597
599
mock_minio_adaptor = MockMinioAdaptor (mocker , files = ['one_minio_entry' ])
598
600
mock_logger = mocker .MagicMock (spec = log_adaptor )
599
- data_adaptor = mocker .MagicMock (spec = DataConverterAdaptor )
601
+ data_adaptor = asyncmock .MagicMock (spec = DataConverterAdaptor )
600
602
data_adaptor .convert_to_awkward .return_value = {'JetPt' : 10 }
601
603
602
604
ds = fe .ServiceXDataset ('localds://mc16_tev:13' ,
@@ -619,7 +621,7 @@ async def test_stream_pandas_data(mocker):
619
621
mock_servicex_adaptor = MockServiceXAdaptor (mocker , "123-456" )
620
622
mock_minio_adaptor = MockMinioAdaptor (mocker , files = ['one_minio_entry' ])
621
623
mock_logger = mocker .MagicMock (spec = log_adaptor )
622
- data_adaptor = mocker .MagicMock (spec = DataConverterAdaptor )
624
+ data_adaptor = asyncmock .MagicMock (spec = DataConverterAdaptor )
623
625
data_adaptor .convert_to_pandas .return_value = {'JetPt' : 10 }
624
626
625
627
ds = fe .ServiceXDataset ('localds://mc16_tev:13' ,
You can’t perform that action at this time.
0 commit comments