map-machine/test/test_direction.py
2020-09-09 23:54:15 +03:00

22 lines
489 B
Python

"""
Test direction processing.
Author: Sergey Vartanov (me@enzet.ru).
"""
import numpy as np
from roentgen.direction import parse_vector
def test_compass_points_1():
assert np.allclose(parse_vector("N"), np.array([0, -1]))
def test_compass_points_2():
root: np.float64 = -np.sqrt(2) / 2
assert np.allclose(parse_vector("NW"), np.array([root, root]))
def test_compass_points_3():
assert np.allclose(
parse_vector("SSW"), np.array([-0.38268343, 0.92387953]))