Add test for main direction.

This commit is contained in:
Sergey Vartanov 2021-06-04 04:08:04 +03:00
parent 0747f279c8
commit 172b48af53

View file

@ -3,7 +3,7 @@ Test direction processing.
"""
import numpy as np
from roentgen.direction import parse_vector
from roentgen.direction import parse_vector, DirectionSet
__author__ = "Sergey Vartanov"
__email__ = "me@enzet.ru"
@ -33,3 +33,11 @@ def test_invalid() -> None:
def test_degree() -> None:
"""Test east direction."""
assert np.allclose(parse_vector("90"), np.array([1, 0]))
def test_main_direction() -> None:
"""Test main direction computing."""
assert DirectionSet("0").is_right() is None
assert DirectionSet("70").is_right() is True
assert DirectionSet("270").is_right() is False
assert DirectionSet("180").is_right() is None