Python arr_mag_to_geo Examples

Example #1

File: test_convert.py Project: UniOfLeicester/plotdarn

def test_convert_array_np_input():
    time = datetime(year=2012, month=6, day=15, hour=22, minute=2)
    input_lat = np.array([north_pole.lat, north_pole.lat])
    input_lon = np.array([north_pole.lon, north_pole.lon])
    res = convert.arr_mag_to_geo(input_lat, input_lon, time)
    expected = np.array([[NP_GLAT, NP_GLAT], [NP_GLON, NP_GLON]])
    np.testing.assert_array_equal(res, expected)

Example #2

File: test_convert.py Project: UniOfLeicester/plotdarn

def test_convert_array_np_input_shape():
    time = datetime(year=2012, month=6, day=15, hour=22, minute=2)
    input_lat = np.array([[north_pole.lat], [north_pole.lat]])
    input_lon = np.array([[north_pole.lon], [north_pole.lon]])
    with pytest.raises(ValueError):
        convert.arr_mag_to_geo(input_lat, input_lon, time)

Example #3

File: test_convert.py Project: UniOfLeicester/plotdarn

def test_convert_array_uneven():
    time = datetime(year=2012, month=6, day=15, hour=22, minute=2)
    with pytest.raises(ValueError):
        convert.arr_mag_to_geo([north_pole.lat, north_pole.lat],
                               [north_pole.lon], time)

Example #4

File: test_convert.py Project: UniOfLeicester/plotdarn

def test_convert_array_single_input():
    time = datetime(year=2012, month=6, day=15, hour=22, minute=2)
    with pytest.raises(TypeError):
        convert.arr_mag_to_geo(north_pole.lat, north_pole.lon, time)

Example #5

File: test_convert.py Project: UniOfLeicester/plotdarn

def test_convert_array():
    time = datetime(year=2012, month=6, day=15, hour=22, minute=2)
    res = convert.arr_mag_to_geo([north_pole.lat], [north_pole.lon], time)
    np.testing.assert_array_equal(res, np.array([[NP_GLAT], [NP_GLON]]))