def test_ephem_sample_same_epochs_returns_same_input(epochs, coordinates, method): unused_plane = Planes.EARTH_EQUATOR ephem = Ephem(coordinates, epochs, unused_plane) result_coordinates = ephem.sample(epochs, method=method) # TODO: Should it return exactly the same? assert_coordinates_allclose(result_coordinates, coordinates, atol_scale=1e-17)
def test_ephem_sample_no_arguments_returns_exactly_same_input( epochs, coordinates, method): unused_plane = Planes.EARTH_EQUATOR ephem = Ephem(coordinates, epochs, unused_plane) result_coordinates = ephem.sample(method=method) # Exactly the same assert np.all(result_coordinates == coordinates)
def test_ephem_sample_scalar_epoch_returns_1_dimensional_coordinates( epochs, coordinates, method): unused_plane = Planes.EARTH_EQUATOR ephem = Ephem(coordinates, epochs, unused_plane) result_coordinates = ephem.sample(epochs[0], method=method) # Exactly the same assert result_coordinates.ndim == 1
def test_ephem_sample_existing_epochs_returns_corresponding_input( epochs, coordinates, method ): unused_plane = Planes.EARTH_EQUATOR ephem = Ephem(coordinates, epochs, unused_plane) result_coordinates = ephem.sample(epochs[::2], method=method) # Exactly the same assert_coordinates_allclose(result_coordinates, coordinates[::2], atol_scale=1e-17)
def test_ephem_sample_scalar_epoch_and_coordinates_returns_exactly_same_input( epochs, coordinates, method): unused_plane = Planes.EARTH_EQUATOR coordinates = coordinates[0].reshape(-1) epochs = epochs[0].reshape(-1) ephem = Ephem(coordinates, epochs, unused_plane) result_coordinates = ephem.sample(epochs[0], method=method) # Exactly the same assert result_coordinates == coordinates