mirror of
https://github.com/enzet/map-machine.git
synced 2025-05-03 20:26:45 +02:00
35 lines
835 B
Python
35 lines
835 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():
|
|
""" Test north direction. """
|
|
assert np.allclose(parse_vector("N"), np.array([0, -1]))
|
|
|
|
|
|
def test_compass_points_2():
|
|
""" Test north-west direction. """
|
|
root: np.float64 = -np.sqrt(2) / 2
|
|
assert np.allclose(parse_vector("NW"), np.array([root, root]))
|
|
|
|
|
|
def test_compass_points_3():
|
|
""" Test south-south-west direction. """
|
|
assert np.allclose(
|
|
parse_vector("SSW"), np.array([-0.38268343, 0.92387953]))
|
|
|
|
|
|
def test_invalid():
|
|
""" Test invalid direction representation string. """
|
|
assert not parse_vector("O")
|
|
|
|
|
|
def test_degree():
|
|
""" Test east direction. """
|
|
assert np.allclose(parse_vector("90"), np.array([1, 0]))
|