Coverage for NeuralTSNE/NeuralTSNE/TSNE/tests/fixtures/dataloader_fixtures.py: 100%

8 statements  

« prev     ^ index     » next       coverage.py v7.8.0, created at 2025-05-18 16:32 +0000

1from unittest.mock import patch 

2 

3import pytest 

4from torch.utils.data import DataLoader, Dataset 

5 

6 

7@pytest.fixture 

8def mock_dataloaders(): 

9 with patch( 

10 "NeuralTSNE.TSNE.ParametricTSNE.parametric_tsne.ParametricTSNE.create_dataloaders", 

11 autospec=True, 

12 ) as mock_create_dataloaders: 

13 mock_create_dataloaders.return_value = DataLoader(Dataset()), DataLoader( 

14 Dataset() 

15 ) 

16 yield mock_create_dataloaders