Coverage for tests / test_fit_source.py: 52%

1# This file is part of multiprofit. 

2# 

3# Developed for the LSST Data Management System. 

4# This product includes software developed by the LSST Project 

5# (https://www.lsst.org). 

6# See the COPYRIGHT file at the top-level directory of this distribution 

7# for details of code ownership. 

8# 

9# This program is free software: you can redistribute it and/or modify 

10# it under the terms of the GNU General Public License as published by 

11# the Free Software Foundation, either version 3 of the License, or 

12# (at your option) any later version. 

13# 

14# This program is distributed in the hope that it will be useful, 

15# but WITHOUT ANY WARRANTY; without even the implied warranty of 

16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

17# GNU General Public License for more details. 

18# 

19# You should have received a copy of the GNU General Public License 

20# along with this program. If not, see <https://www.gnu.org/licenses/>. 

21 

22import lsst.gauss2d.fit as g2f 

23from lsst.multiprofit import ( 

24 ComponentGroupConfig, GaussianComponentConfig, ModelConfig, ModelFitConfig, SourceConfig, 

25) 

26from lsst.multiprofit.fitting.fit_source import CatalogSourceFitterConfig, CatalogSourceFitterConfigData 

27from lsst.multiprofit.utils import get_params_uniq 

28import pytest 

29 

30 

31@pytest.fixture(scope="module") 

32def channels() -> tuple[g2f.Channel]: 

33 channels = tuple(g2f.Channel.get(band) for band in ("R", "G", "B")) 

34 return channels 

35 

36 

37@pytest.fixture(scope="module") 

38def fitter_config() -> CatalogSourceFitterConfig: 

39 config = CatalogSourceFitterConfig( 

40 config_fit=ModelFitConfig(), 

41 config_model=ModelConfig( 

42 sources={ 

43 "": SourceConfig( 

44 component_groups={ 

45 "": ComponentGroupConfig( 

46 components_gauss=( 

47 { 

48 "gauss": GaussianComponentConfig() 

49 } 

50 ), 

51 ) 

52 } 

53 ), 

54 }, 

55 ), 

56 fit_psmodel_final=True, 

57 ) 

58 return config 

59 

60 

61@pytest.fixture(scope="module") 

62def fitter_config_data(channels, fitter_config) -> CatalogSourceFitterConfigData: 

63 config_data = CatalogSourceFitterConfigData(channels=channels, config=fitter_config) 

64 return config_data 

65 

66 

67def test_fitter_config_data(fitter_config_data): 

68 parameters = fitter_config_data.parameters 

69 assert len(parameters) > 0 

70 sources, priors = fitter_config_data.sources_priors 

71 same = (p1 is p2 for p1, p2 in zip(parameters.values(), get_params_uniq(sources[0], fixed=False))) 

72 assert all(same)