Example #1
0
def test_normalize_ang_parameters():
    '''
	Test that the correct parameters are used.
	'''

    with pytest.raises(Exception):
        _normalize_ang(ra=ra, points=points)
Example #2
0
def test_normalize_ang_int_arrays():
    '''
	Test handling of non-floating point arrays.
	'''
    # ra is np.int32 and we asked not to copy the values
    with pytest.raises(Exception):
        _normalize_ang(ra_int, dec, copy=False)
Example #3
0
def test_normalize_ang_points_array():
    '''
	Test passing in a 'points' array.
	'''

    # create a 2D array; only use the first and last points
    points = np.squeeze(np.dstack((np.copy(ra).astype(np.double), dec)))
    _normalize_ang(points=points)

    # change shape of output
    points_norm = np.squeeze(np.dstack((ra_norm, dec_norm)))

    assert_allclose(
        points[:, 0],
        points_norm[:, 0],
        err_msg="converting a 'points' array failed; ra didn't match")
    assert_allclose(
        points[:, 1],
        points_norm[:, 1],
        err_msg="converting a 'points' array failed; dec didn't match")

    assert (points[:, 0] >= 0).all() and (
        points[:, 0] <= 360).all(), "ra values out of [0,360]!!"
    assert (points[:, 1] >= -90).all() and (
        points[:, 1] <= 90).all(), "dec values out of [-90,90]!!"
Example #4
0
def test_normalize_ang_mismatching_array_lengths():
    '''
	Test mismatching array lengths.
	'''
    ra = np.array([1, 2, 3, 4], dtype=np.double)
    dec = np.array([53, 654], dtype=np.double)

    with pytest.raises(Exception):
        _normalize_ang(ra, dec)
Example #5
0
def test_normalize_ang_float_arrays():
    '''
	Test float arrays when copied.
	'''
    ra = np.array([451, 23, 54, -5], dtype=np.float32)
    dec = np.array([12, 54, -95, 5], dtype=np.float32)

    ra_in_id = id(ra)
    dec_in_id = id(dec)

    ra_out, dec_out = _normalize_ang(ra, dec, copy=True)

    assert_allclose(
        ra_out,
        np.array([91., 23., 234., 355.], dtype=np.float32),
        err_msg="ra doesn't match when input is float array and copy requested"
    )
    assert_allclose(
        dec_out,
        np.array([12., 54., -85., 5.], dtype=np.float32),
        err_msg="dec doesn't match when input is float array and copy requested"
    )

    assert ra_in_id != id(ra_out), "ra copy requested, but original modified"
    assert dec_in_id != id(
        dec_out), "dec copy requested, but original modified"
Example #6
0
def test_normalize_ang_scalars(ra, dec, ra_norm, dec_norm):
    '''
	Test that scalar values work.
	'''
    ra_out, dec_out = _normalize_ang(ra, dec)
    assert_approx_equal(ra_out,
                        ra_norm,
                        err_msg="RA scalar values did not match.")
    assert_approx_equal(dec_out,
                        dec_norm,
                        err_msg="dec scalar values did not match.")
Example #7
0
def test_normalize_ang_individual_arrays():
    '''
	Test _normalize_ang when given individual ra, dec arrays.
	'''
    ra_id = id(ra)
    dec_id = id(dec)

    _normalize_ang(ra, dec)

    assert_allclose(ra,
                    ra_norm,
                    err_msg="The RA array did not normalize correctly.")
    assert_allclose(dec,
                    dec_norm,
                    err_msg="The dec array did not normalize correctly.")

    # the arrays should not have been copied
    assert ra_id == id(ra), "the ra array was unexpectedly copied"
    assert dec_id == id(dec), "the dec array was unexpectedly copied"

    # ultimately the point...
    assert (ra >= 0).all() and (ra <= 360).all(), "ra values out of [0,360]!!"
    assert (dec >= -90).all() and (dec <=
                                   90).all(), "dec values out of [-90,90]!!"
Example #8
0
def test_normalize_ang_points_return_on_copy():
    '''
	Test that a single 'points' array is returned when one points array is provided.
	'''
    points = np.squeeze(
        np.dstack((np.array([451, 23, 54, -5], dtype=np.float32),
                   np.array([12, 54, -95, 5], dtype=np.float32))))

    points_id = id(points)

    points_out = _normalize_ang(points=points, copy=True)

    points_norm = np.array([[91., 12.], [23., 54.], [234., -85.], [355., 5.]],
                           dtype=np.float32)

    assert_allclose(points_out,
                    points_norm,
                    err_msg="points copy test doesn't match")

    assert points_id != id(
        points_out), "the original 'points' array was modified, not copied"