* Included tests for "Fleet". #41

* Included test for "Vehicle.as_dict()".

tests/test_models/test_fleets.py

@@ -0,0 +1,31 @@
+import unittest
+
+import jinete as jit
+
+from tests.utils import (
+    generate_vehicles,
+)
+
+
+class TestFleet(unittest.TestCase):
+
+    def test_construction(self):
+        vehicles = generate_vehicles(3)
+        fleet = jit.Fleet(vehicles)
+
+        self.assertIsInstance(fleet, jit.Fleet)
+        self.assertEqual(fleet.vehicles, vehicles)
+
+    def test_as_dict(self):
+        vehicles = generate_vehicles(3)
+        fleet = jit.Fleet(vehicles)
+
+        vehicles_str = ', '.join(str(vehicle) for vehicle in vehicles)
+        expected = {
+            'vehicles': f'{{{vehicles_str}}}'
+        }
+        self.assertEqual(expected, fleet.as_dict())
+
+
+if __name__ == '__main__':
+    unittest.main()

tests/test_models/test_vehicles.py

@@ -85,6 +85,24 @@ def test_vehicle_with_timeout(self):
         self.assertEqual(vehicle.timeout, timeout)
         self.assertIsInstance(vehicle.uuid, UUID)
 
+    def test_as_dict(self):
+        initial = generate_one_position()
+        identifier = str(0)
+        vehicle = jit.Vehicle(identifier, initial)
+
+        expected = {
+            'initial': vehicle.initial,
+            'final': vehicle.final,
+            'capacity': vehicle.capacity,
+            'earliest': vehicle.earliest,
+            'timeout': vehicle.timeout,
+            'route_timeout': vehicle.route_timeout,
+            'trip_timeout': vehicle.trip_timeout,
+            'latest': vehicle.latest,
+            'uuid': vehicle.uuid,
+        }
+
+        self.assertEqual(expected, vehicle.as_dict())
 
 if __name__ == '__main__':
     unittest.main()