thib8956/sumo-emissions
1
0
Fork 0
mirror of https://github.com/Ahp06/SUMO_Emissions.git synced 2024-11-24 04:26:29 +00:00
Code Issues Releases Wiki Activity

Update tests

Browse Source
This commit is contained in:
Thibaud Gasser 2018-12-13 21:02:07 +01:00
parent 576112dd6a
commit ceea79035d
1 changed files with 49 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
tests/configurator_tests.py
View File

@ -2,37 +2,17 @@ import os
import shutil import shutil
import tempfile import tempfile
import unittest import unittest
from configurator import configurator from configurator import configurator
# Absolute path of the directory the script is in
SCRIPTDIR = os.path.dirname(__file__)
class ConfiguratorTests(unittest.TestCase): class TemplateTests(unittest.TestCase):
def setUp(self): def setUp(self):
self.sim_path = tempfile.mkdtemp()
self.sim_name = 'test_simulation' self.sim_name = 'test_simulation'
self.sim_path = '/test_simulation'
def tearDown(self): self.log_path = '/test_simulation/log'
shutil.rmtree(self.sim_path)
def test_generate_scenario(self):
osm_file = os.path.abspath('sample.osm')
configurator.generate_scenario(osm_file, self.sim_path, self.sim_name)
self.assert_is_dir(self.sim_path)
# self.assert_is_dir(os.path.join(self.sim_path, 'log'))
generated_files = [
f'{self.sim_name}.poly.xml',
f'{self.sim_name}.net.xml'
]
for f in generated_files:
self.assert_is_file(os.path.join(self.sim_path, f))
def test_generate_mobility(self):
# The scenario must be generated before the mobility
osm_file = os.path.abspath('sample.osm')
configurator.generate_scenario(osm_file, self.sim_path, self.sim_name)
configurator.generate_mobility(self.sim_path, self.sim_name)
trips_file = os.path.join(self.sim_path, f'{self.sim_name}.trips.xml')
self.assert_is_file(trips_file)
def test_load_netconvert_template(self): def test_load_netconvert_template(self):
tree = configurator.load_netconvert_template('test.osm', 'test_simulation') tree = configurator.load_netconvert_template('test.osm', 'test_simulation')
@ -40,13 +20,21 @@ class ConfiguratorTests(unittest.TestCase):
self.assertEqual(tree.find('output/output-file').get('value'), f'{self.sim_name}.net.xml') self.assertEqual(tree.find('output/output-file').get('value'), f'{self.sim_name}.net.xml')
self.assertEqual(tree.find('report/log').get('value'), f'{self.sim_name}.netconvert.log') self.assertEqual(tree.find('report/log').get('value'), f'{self.sim_name}.netconvert.log')
def test_load_sumoconfig_template(self): def test_load_sumoconfig_template_default(self):
tree = configurator.load_sumoconfig_template(self.sim_name) tree = configurator.load_sumoconfig_template(self.sim_name)
self.assertEqual(tree.find('input/net-file').get('value'), f'{self.sim_name}.net.xml') self.assertEqual(tree.find('input/net-file').get('value'), f'{self.sim_name}.net.xml')
self.assertEqual(tree.find('input/route-files').get('value'), f'{self.sim_name}.rou.xml') self.assertEqual(tree.find('input/route-files').get('value'), f'{self.sim_name}.rou.xml')
self.assertEqual(tree.find('input/additional-files').get('value'), f'{self.sim_name}.poly.xml') self.assertEqual(tree.find('input/additional-files').get('value'), f'{self.sim_name}.poly.xml')
self.assertEqual(tree.find('report/log').get('value'), f'{self.sim_name}.log') self.assertEqual(tree.find('report/log').get('value'), f'{self.sim_name}.log')
def test_load_sumoconfig_template_with_routefiles(self):
routefiles = (f'{self.sim_name}.bus.rou.xml', f'{self.sim_name}.passenger.rou.xml')
tree = configurator.load_sumoconfig_template(self.sim_name, routefiles)
self.assertEqual(tree.find('input/net-file').get('value'), f'{self.sim_name}.net.xml')
self.assertEqual(tree.find('input/route-files').get('value'), ','.join(routefiles))
self.assertEqual(tree.find('input/additional-files').get('value'), f'{self.sim_name}.poly.xml')
self.assertEqual(tree.find('report/log').get('value'), f'{self.sim_name}.log')
def test_load_polyconvert_template(self): def test_load_polyconvert_template(self):
tree = configurator.load_polyconvert_template( tree = configurator.load_polyconvert_template(
osm_file=f'{self.sim_name}.osm', osm_file=f'{self.sim_name}.osm',
@ -59,6 +47,40 @@ class ConfiguratorTests(unittest.TestCase):
self.assertEqual(tree.find('output/output-file').get('value'), f'{self.sim_name}.poly.xml') self.assertEqual(tree.find('output/output-file').get('value'), f'{self.sim_name}.poly.xml')
self.assertEqual(tree.find('report/log').get('value'), f'{self.sim_name}.polyconvert.log') self.assertEqual(tree.find('report/log').get('value'), f'{self.sim_name}.polyconvert.log')
class GenerationTests(unittest.TestCase):
def setUp(self):
self.base_path = tempfile.mkdtemp()
self.sim_name = 'test_simulation'
self.sim_path = os.path.join(self.base_path, self.sim_name)
self.log_path = os.path.join(self.sim_name, 'log')
def tearDown(self):
shutil.rmtree(self.base_path)
def test_generate_scenario(self):
osm_file = os.path.join(SCRIPTDIR, 'sample.osm')
configurator.generate_scenario(osm_file, self.sim_path, self.sim_name)
self.assert_is_dir(self.sim_path)
generated_files = [
f'{self.sim_name}.poly.xml',
f'{self.sim_name}.net.xml'
]
for f in generated_files:
self.assert_is_file(os.path.join(self.sim_path, f))
def test_generate_mobility(self):
# The scenario must be generated before the mobility
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)
self.assert_is_file(trips_file)
for f in routefiles:
self.assert_is_file(os.path.join(self.sim_path, f))
def assert_exists(self, path): def assert_exists(self, path):
self.assertTrue(os.path.exists(path), msg=f'{path} does not exist') self.assertTrue(os.path.exists(path), msg=f'{path} does not exist')