scenario_output_test.py

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)