Python Model.get_number_of_point_sources Examples

Example #1

File: test_model.py Project: sybenzvi/astromodels

def test_constructor_with_many_point_sources():

    # Test with 200 point sources

    many_p_sources = map(lambda x:_get_point_source("pts_source%i" %x), range(200))

    m = Model(*many_p_sources)

    assert m.get_number_of_point_sources()==200

Example #2

File: test_model.py Project: sybenzvi/astromodels

def test_constructor_with_mix():

    many_p_sources = map(lambda x: _get_point_source("pts_source%i" % x), range(200))

    many_e_sources = map(lambda x: _get_extended_source("ext_source%i" % x), range(200))

    many_part_sources = map(lambda x: _get_particle_source("part_source%i" % x), range(200))

    all_sources = []
    all_sources.extend(many_p_sources)
    all_sources.extend(many_e_sources)
    all_sources.extend(many_part_sources)

    m = Model(*all_sources)

    assert m.get_number_of_point_sources() == 200
    assert m.get_number_of_extended_sources() == 200
    assert m.get_number_of_particle_sources() == 200