import unittest from vadere_analysis_tool import ScenarioOutput class ScenarioOutputTests(unittest.TestCase): def test_output_dir(self): out = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-42-54.117") self.assertEqual(out.output_dir, "testData/s2ucre/output/b_2018-11-16_13-42-54.117") def test_scenario_dict(self): out = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-42-54.117") self.assertEqual(out.scenario['name'], 'bridge_coordinates_kai') def test_invalid_scenario_dict(self): self.assertRaises(FileNotFoundError, ScenarioOutput, "testData/s2ucre/output/invalid") def test_scenario_output_dict(self): out = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-42-54.117") self.assertEqual(len(out.files), 3) def test_settattr_works(self): out = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-42-54.117") self.assertTrue('df_overlapCount_txt' in dir(out.named_files)) self.assertTrue('df_overlaps_csv' in dir(out.named_files)) self.assertTrue('df_postvis_trajectories' in dir(out.named_files)) self.assertTrue('info_overlapCount_txt' in dir(out.named_files)) self.assertTrue('info_overlaps_csv' in dir(out.named_files)) self.assertTrue('info_postvis_trajectories' in dir(out.named_files)) def test_offset(self): out = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-42-54.117") self.assertEqual([564280.0,5933391.0], out.get_bound_offset()) def test_get_scenario_name(self): out = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-42-54.117") self.assertEqual('bridge_coordinates_kai', out.get_scenario_name()) def test_scenario_md5sum(self): out_1 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-42-54.117") out_2 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-57-42.894") out_3 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-43-08.160") out_4 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-57-58.997") out_5 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_13-46-13.488") out_6 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_14-01-07.289") out_7 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-49-31.248") out_8 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_14-04-19.882") out_9 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_13-49-48.269") out_11 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_14-04-35.721") out_12 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_13-53-04.555") out_13 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_14-08-14.82") out_14 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_13-56-18.210") out_15 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_14-11-36.817") out_16 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-56-34.297") out_17 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_14-11-53.469") out_18 = ScenarioOutput("testData/s2ucre/output/empty_2018-11-16_13-56-50.397") out_19 = ScenarioOutput("testData/s2ucre/output/empty_2018-11-16_14-12-09.609") self.assertEqual('774fc8aaaf93ee003709626bb4c0db1f', out_1.scenario_hash) self.assertEqual('774fc8aaaf93ee003709626bb4c0db1f', out_2.scenario_hash) self.assertEqual('7be534d60a54f00afa1f9244466e8dcf', out_3.scenario_hash) self.assertEqual('7be534d60a54f00afa1f9244466e8dcf', out_4.scenario_hash) self.assertEqual('2b706a0ed12842f7ecfcc1b966b2edcc', out_5.scenario_hash) self.assertEqual('2b706a0ed12842f7ecfcc1b966b2edcc', out_6.scenario_hash) self.assertEqual('925007d0e16f698f404ede36528bbf49', out_7.scenario_hash) self.assertEqual('925007d0e16f698f404ede36528bbf49', out_8.scenario_hash) self.assertEqual('dce2cc64adbcd4520b5217afb64061e8', out_9.scenario_hash) self.assertEqual('dce2cc64adbcd4520b5217afb64061e8', out_11.scenario_hash) self.assertEqual('9bd23355bdc8e34e6766894bca2b469c', out_12.scenario_hash) self.assertEqual('9bd23355bdc8e34e6766894bca2b469c', out_13.scenario_hash) self.assertEqual('737fc0ed9b1ff0281043facf5d5a26da', out_14.scenario_hash) self.assertEqual('737fc0ed9b1ff0281043facf5d5a26da', out_15.scenario_hash) self.assertEqual('3405c30c7449d7b48d1ca21947433960', out_16.scenario_hash) self.assertEqual('3405c30c7449d7b48d1ca21947433960', out_17.scenario_hash) self.assertEqual('cd909753844587db821d1222e31d80d1', out_18.scenario_hash) self.assertEqual('cd909753844587db821d1222e31d80d1', out_19.scenario_hash) def test_trajectories_md5sum(self): out_1 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-42-54.117") out_2 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-57-42.894") out_3 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-43-08.160") out_4 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-57-58.997") out_5 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_13-46-13.488") out_6 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_14-01-07.289") out_7 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-49-31.248") out_8 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_14-04-19.882") out_9 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_13-49-48.269") out_11 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_14-04-35.721") out_12 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_13-53-04.555") out_13 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_14-08-14.82") out_14 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_13-56-18.210") out_15 = ScenarioOutput( "testData/s2ucre/output/b_2018-11-16_14-11-36.817") out_16 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_13-56-34.297") out_17 = ScenarioOutput("testData/s2ucre/output/b_2018-11-16_14-11-53.469") out_18 = ScenarioOutput("testData/s2ucre/output/empty_2018-11-16_13-56-50.397") out_19 = ScenarioOutput("testData/s2ucre/output/empty_2018-11-16_14-12-09.609") self.assertEqual('c1968d8bc56e1b89402dfccf55d7a385', out_1.trajectories_hash) self.assertEqual('c1968d8bc56e1b89402dfccf55d7a385', out_2.trajectories_hash) self.assertEqual('03dbf9f8ec447146faadafb5f69521e9', out_3.trajectories_hash) self.assertEqual('03dbf9f8ec447146faadafb5f69521e9', out_4.trajectories_hash) self.assertEqual('2188b6f97a7e14f8da8ca8fdfe34a817', out_5.trajectories_hash) self.assertEqual('2188b6f97a7e14f8da8ca8fdfe34a817', out_6.trajectories_hash) self.assertEqual('50f5506cac5ebf6dec8e082947a3de52', out_7.trajectories_hash) self.assertEqual('50f5506cac5ebf6dec8e082947a3de52', out_8.trajectories_hash) self.assertEqual('e9da9eb826f4d12e98a405941e8f8377', out_9.trajectories_hash) self.assertEqual('e9da9eb826f4d12e98a405941e8f8377', out_11.trajectories_hash) self.assertEqual('2257e7bce72e06ab0d7ad9e67eb8adcf', out_12.trajectories_hash) self.assertEqual('2257e7bce72e06ab0d7ad9e67eb8adcf', out_13.trajectories_hash) self.assertEqual('cdfb67ce2a689b42d168a023803fe8ef', out_14.trajectories_hash) self.assertEqual('cdfb67ce2a689b42d168a023803fe8ef', out_15.trajectories_hash) self.assertEqual('4dfc3817db68a069f7180680d249d550', out_16.trajectories_hash) self.assertEqual('4dfc3817db68a069f7180680d249d550', out_17.trajectories_hash) self.assertEqual('6f077db2b6af4e022f970cbe4ff3b1f8', out_18.trajectories_hash) self.assertEqual('6f077db2b6af4e022f970cbe4ff3b1f8', out_19.trajectories_hash)