elc_test.py

"""Unit test for the elcpy module."""

import unittest
import wsdotelc


class Test_unittest(unittest.TestCase):
    def setUp(self):
        self.elc = wsdotelc.Elc()

    def test_routes(self):
        """Test the retrieval of `elcpy.Elc.routes`."""
        routes = self.elc.routes
        self.assertTrue(isinstance(routes, dict), "Returned routes object is a dict.")
        self.assertTrue(
            isinstance(self.elc._routes, dict), "Route dict has been cached."
        )

    def test_find_route_locations(self):
        """Test the `elcpy.Elc.find_route_locations` function."""
        # locations = (elcpy.RouteLocation(route="005", arm=5, reference_date="12/31/2013"))
        # Create a set of locations.
        locations = (wsdotelc.RouteLocation(Route="005", Arm=5),)
        out_locations = self.elc.find_route_locations(locations, "12/31/2013")
        self.assertEqual(len(out_locations), 1, "Result has single element.")
        self.assertIsInstance(
            out_locations[0],
            wsdotelc.RouteLocation,
            "The first element in the returned array is an `elcpy.RouteLocation`.",
        )

    def test_find_nearest_route_location(self):
        """Test the `elcpy.Elc.find_dearest_route_locations` function."""
        points = [1087403.28714286, 136623.00728571415]
        out_locations = self.elc.find_nearest_route_locations(
            points, "12/31/2013", 200, 2927
        )
        self.assertEqual(
            1,
            len(out_locations),
            "Input and output loctions should have the same number of elements.",
        )
        self.assertIsInstance(
            out_locations[0],
            wsdotelc.RouteLocation,
            "The first element in the returned array is an `elcpy.RouteLocation`.",
        )
        self.assertListEqual(
            [out_locations[0].Route, out_locations[0].Arm],
            ["005", 5],
            "Test for expected Route ID and ARM values.",
        )

    # def test_A(self):
    #    self.fail("Not implemented")


if __name__ == "__main__":
    unittest.main()
    # suite = unittest.TestLoader().loadTestsFromTestCase(Test_unittest)
    # unittest.TextTestRunner(verbosity=2).run(suite)