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

1from typing import Any 

2 

3import pytest 

4 

5from NeuralTSNE.TSNE.NeuralNetwork import NeuralNetwork 

6 

7 

8@pytest.fixture 

9def neural_network_params( 

10 request: type[pytest.FixtureRequest], 

11) -> dict[str, Any]: 

12 initial_features, n_components, multipliers, *args = request.param 

13 args_keys = ["pre_filled_layers"] 

14 

15 return { 

16 "initial_features": initial_features, 

17 "n_components": n_components, 

18 "multipliers": multipliers, 

19 } | dict(zip(args_keys, args)) 

20 

21 

22@pytest.fixture 

23def neural_network(neural_network_params: dict[str, Any]) -> NeuralNetwork: 

24 return NeuralNetwork(**neural_network_params)