import unittest
from ..points import NamedPosition, StraightSegment, Position, NP


class NamedPositionTests(unittest.TestCase):
    def test_simple(self):
        namedpos = NamedPosition(Position(3, 42), ["testname"])
        self.assertEqual(namedpos.x, 3)
        self.assertEqual(namedpos.y, 42)
        self.assertEqual(namedpos.names, ["testname"])

    def test_nonstr(self):
        with self.assertRaises(TypeError):
            NamedPosition(Position(0, 1), ["name1", "name2", 2])

    def test_first(self):
        namedpos = NamedPosition(Position(0, 1), ["name1", "name2"])
        self.assertEqual(namedpos.first, "name1")


class StraightSegmentTests(unittest.TestCase):
    def test_straight_segment(self):
        ss = StraightSegment(StraightSegment.Type.S, [NP(0, 0)])
        ss.append(NP(0, 1))
        ss.append(NP(0, 2))
        self.assertEqual(ss.d, StraightSegment.Type.V)

        self.assertEqual(len(ss), 3)

    def test_segment_append(self):
        ss = StraightSegment(StraightSegment.Type.S, [NP(0, 0)])
        ss.append(NP(0, 1))
        # Add point in different direction
        with self.assertRaises(AssertionError):
            ss.append(NP(1, 0))

    def test_names(self):
        ss = StraightSegment(StraightSegment.Type.S, [NP(0, 0, "n1", "n3")])
        self.assertEqual(sorted(ss.names), ["n1", "n3"])

        ss.append(NP(0, 1, "n2", "n4"))
        self.assertEqual(sorted(ss.names), ["n1", "n2", "n3", "n4"])


class Decompose(unittest.TestCase):
    def test_decompose_points(self):
        pass