Python read_density_file Examples

Example #1

File: test_ascii_readers.py Project: Chrischanmarc/tardis

def test_ascii_reader_density_boundaries(v_inner_boundary, v_outer_boundary, actual_v_inner, actual_v_outer,
                                         inner_index, outer_index):
    v_inner, v_outer, mean_densities, inner_boundary_index, outer_boundary_index = \
        io.read_density_file(data_path('artis_model.dat'), 'artis', 19 * u.day, v_inner_boundary, v_outer_boundary)

    assert inner_boundary_index == inner_index
    assert outer_boundary_index == outer_index

    if not np.isnan(actual_v_inner):
        assert_almost_equal(v_inner[0], actual_v_inner)

    if not np.isnan(actual_v_outer):
        assert_almost_equal(v_outer[-1], actual_v_outer)

Example #2

def test_ascii_reader_density_boundaries(v_inner_boundary, v_outer_boundary,
                                         actual_v_inner, actual_v_outer,
                                         inner_index, outer_index):
    v_inner, v_outer, mean_densities, inner_boundary_index, outer_boundary_index = \
        io.read_density_file(data_path('artis_model.dat'), 'artis', 19 * u.day, v_inner_boundary, v_outer_boundary)

    assert inner_boundary_index == inner_index
    assert outer_boundary_index == outer_index

    if not np.isnan(actual_v_inner):
        assert_almost_equal(v_inner[0], actual_v_inner)

    if not np.isnan(actual_v_outer):
        assert_almost_equal(v_outer[-1], actual_v_outer)

Example #3

def test_ascii_reader_density_boundaries(v_inner_boundary, v_outer_boundary, actual_v_inner, actual_v_outer,
                                         inner_index, outer_index):
    v_inner, v_outer, mean_densities, inner_boundary_index, outer_boundary_index = \
        io.read_density_file(data_path('artis_model.dat'), 'artis', 19 * u.day, v_inner_boundary, v_outer_boundary)

    assert inner_boundary_index == inner_index
    assert outer_boundary_index == outer_index

    if not np.isnan(actual_v_inner):
        npt.assert_allclose(v_inner[0].value,
                            actual_v_inner.to(v_inner[0].unit).value)

    if not np.isnan(actual_v_outer):
        npt.assert_allclose(v_outer[-1].value,
                            actual_v_outer.to(v_outer[-1].unit).value)

Example #4

def test_ascii_reader_invalid_volumes():
    with pytest.raises(io.model_reader.ConfigurationError):
        io.read_density_file(data_path('invalid_artis_model.dat'), 'artis')

Example #5

def test_ascii_reader_invalid_volumes():

    with pytest.raises(io.model_reader.ConfigurationError):
        v_inner, v_outer, mean_densities, inner_boundary_index, outer_boundary_index = \
            io.read_density_file(data_path('invalid_artis_model.dat'), 'artis', 19 *
                                 u.day, 0.0 * u.km/u.s, np.inf * u.km/u.s)

Example #6

File: test_ascii_readers.py Project: tardis-sn/tardis

def test_ascii_reader_invalid_volumes():
    with pytest.raises(io.model_reader.ConfigurationError):
        io.read_density_file(data_path('invalid_artis_model.dat'), 'artis')