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)
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)
def test_ascii_reader_invalid_volumes(): with pytest.raises(io.model_reader.ConfigurationError): io.read_density_file(data_path('invalid_artis_model.dat'), 'artis')
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)