thib8956/sumo-emissions
1
0
Fork 0
mirror of https://github.com/Ahp06/SUMO_Emissions.git synced 2025-07-01 21:55:44 +00:00
Code Issues Releases Wiki Activity

Refactor code

Browse Source
This commit is contained in:
Thibaud Gasser
2018-12-10 17:59:26 +01:00
parent 8f69983a80
commit dc4444a24e
3 changed files with 50043 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
tests/configurator_tests.py
View File

@ -4,6 +4,7 @@ import tempfile
import unittest
from configurator import configurator
class ConfiguratorTests(unittest.TestCase):
def setUp(self):
@ -14,11 +15,11 @@ class ConfiguratorTests(unittest.TestCase):
shutil.rmtree(self.sim_path)
def test_generate_scenario(self):
configurator.generate_scenario(self.sim_path, self.sim_name)
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'))
# self.assert_is_dir(os.path.join(self.sim_path, 'log'))
generated_files = [
f'{self.sim_name}.sumocfg',
f'{self.sim_name}.poly.xml',
f'{self.sim_name}.net.xml'
]
@ -27,11 +28,37 @@ class ConfiguratorTests(unittest.TestCase):
def test_generate_mobility(self):
# The scenario must be generated before the mobility
configurator.generate_scenario(self.sim_path, self.sim_name)
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):
tree = configurator.load_netconvert_template('test.osm', 'test_simulation')
self.assertEqual(tree.find('input/osm-files').get('value'), 'test.osm')
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')
def test_load_sumoconfig_template(self):
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/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('report/log').get('value'), f'{self.sim_name}.log')
def test_load_polyconvert_template(self):
tree = configurator.load_polyconvert_template(
osm_file=f'{self.sim_name}.osm',
type_file='typemap/test.typ.xml',
scenario_name=f'{self.sim_name}'
)
self.assertEqual(tree.find('input/osm-files').get('value'), f'{self.sim_name}.osm')
self.assertEqual(tree.find('input/net-file').get('value'), f'{self.sim_name}.net.xml')
self.assertEqual(tree.find('input/type-file').get('value'), 'typemap/test.typ.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')
def assert_exists(self, path):
self.assertTrue(os.path.exists(path), msg=f'{path} does not exist')

49966
tests/sample.osm Normal file
View File

File diff suppressed because it is too large Load Diff