Python get_model_wrapperの例

コード例 #1

ファイル: test_cli.py プロジェクト: peterdfields/stdpopsim

 def test_bad_model(self):
     species = stdpopsim.get_species("HomSap")
     with mock.patch("stdpopsim.cli.exit", autospec=True) as mocked_exit:
         cli.get_model_wrapper(species, "XXX")
         mocked_exit.assert_called_once_with(
             "DemographicModel 'HomSap/XXX' not in catalog"
         )

コード例 #2

ファイル: test_cli.py プロジェクト: fbaumdicker/stdpopsim

 def test_bad_model(self):
     species = stdpopsim.get_species("HomSap")
     with mock.patch("stdpopsim.cli.exit", autospec=True) as mocked_exit:
         cli.get_model_wrapper(species, "XXX")
         available_models = ", ".join([dm.id for dm in species.demographic_models])
         mocked_exit.assert_called_once_with(
             f"DemographicModel 'HomSap/XXX' not in catalog ({available_models})"
         )