From c31a83d1d85b77bddb4b820bb40886bfa2ba409e Mon Sep 17 00:00:00 2001
From: Thibaud <thibaud.gasser@gmx.com>
Date: Wed, 30 Jan 2019 15:56:51 +0100
Subject: [PATCH] Fix tests with vehicle classes

---
 tests/configurator_tests.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/configurator_tests.py b/tests/configurator_tests.py
index ea7e017..1552ff9 100644
--- a/tests/configurator_tests.py
+++ b/tests/configurator_tests.py
@@ -86,7 +86,8 @@ class GenerationTests(unittest.TestCase):
         osm_file = os.path.join(SCRIPTDIR, 'sample.osm')
         trips_file = os.path.join(self.sim_path, f'{self.sim_name}.trips.xml')
         configurator.generate_scenario(osm_file, self.sim_path, self.sim_name)
-        routefiles = configurator.generate_mobility(self.sim_path, self.sim_name, vclasses=('passenger', 'truck'))
+        classes = {'passenger': 10, 'truck': 1}
+        routefiles = configurator.generate_mobility(self.sim_path, self.sim_name, vclasses=classes)
 
         self.assert_is_file(trips_file)
         for f in routefiles: