development branch

Author: r4m-juan

examples/vehicles/get_vehicles.py

@@ -9,19 +9,21 @@ def main(api_key):
 
     vehicles = route4me.vehicles
     response = vehicles.get_vehicles()
-    if type(response) == dict and 'errors' in response.keys():
-        print('. '.join(response.get('errors')))
+    if isinstance(response, dict) and "errors" in response.keys():
+        print(". ".join(response.get("errors")))
     else:
         for vehicle in response:
-            print('Vehicle ID: {0}\tVehicle Alias: {1}'.format(
-                vehicle.get('vehicle_id'),
-                vehicle.get('vehicle_alias')
-            ))
+            print(
+                "Vehicle ID: {0}\tVehicle Alias: {1}".format(
+                    vehicle.get("vehicle_id"), vehicle.get("vehicle_alias")
+                )
+            )
 
 
-if __name__ == '__main__':
-    parser = argparse.ArgumentParser(description='Get Vehicles')
-    parser.add_argument('--api_key', dest='api_key', help='Route4Me API KEY',
-                        type=str, required=True)
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Get Vehicles")
+    parser.add_argument(
+        "--api_key", dest="api_key", help="Route4Me API KEY", type=str, required=True
+    )
     args = parser.parse_args()
     main(args.api_key)

tests/test_optimizations.py

@@ -124,8 +124,7 @@ def test_tsp_optimization(self):
             alias='Verizon Wireless',
             time=0
         )
-        response = self.route4me.run_optimization()
-        self.assertEqual(4, response.get('state'))
+        self.assertEqual(9, len(optimization.data['addresses']))
 
 
 if __name__ == '__main__':