def test_getcosIm_scalar(): assert_allclose(helpers.getcosIm(60), 0.27735009811261463) assert_allclose(helpers.getcosIm(10), 0.94307531289434765) assert type(helpers.getcosIm(60)) != np.ndarray
def test_getcosIm_2Darray(): assert_allclose(helpers.getcosIm([[60, 10], [60, 10]]), [[0.27735009811261463, 0.94307531289434765], [0.27735009811261463, 0.94307531289434765]])