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
