lib/test/test_util.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14

import os

import numpy as np
from sphere import vector

ROOT_DIR = os.path.join(os.path.dirname(__file__), 'data')

def get_point_set(density=25):
    points = []
    for i in np.linspace(-85, 85, density, True):
        for j in np.linspace(-180, 180, np.cos(np.deg2rad(i)) * density):
            points.append([j, i])
    points = np.asarray(points)
    return np.dstack(vector.radec_to_vector(points[:,0], points[:,1]))[0]