Example #1
0
    def test_multiple_threads(self):
        sim = Simulation(5, testing=True)
        sim.start()
        sim.join()

        # 10 days for each iteration, so 50 reports total
        self.assertEqual(DailyControls.objects.count(), 50)
Example #2
0
    def test_multiple_threads(self):
        sim = Simulation(5)
        sim.start()
        sim.join()

        # 4 days for each iteration, so 20 reports total
        self.assertEqual(DailyReport.objects.count(), 20)
Example #3
0
    def test_map_zip_no_output(self):
        settings = OutputSettings.objects.first()
        settings.save_map_output = False
        settings.save()
        close_old_connections()
        file_name = os.path.join(self.scenario_directory, 'Roundtrip_test Map Output.zip')
        folder_name = os.path.join(self.scenario_directory, 'Map')

        sim = Simulation(1, testing=True)
        sim.start()
        sim.join()

        self.assertFalse(os.access(file_name, os.F_OK))
Example #4
0
    def test_map_zip_with_output(self):
        settings = OutputSettings.objects.first()
        settings.save_daily_unit_states = True
        settings.save_map_output = True
        settings.save()
        close_old_connections()
        file_name = os.path.join(self.scenario_directory, 'Roundtrip_test Map Output.zip')
        folder_name = os.path.join(self.scenario_directory, 'Map')

        sim = Simulation(1, testing=True)
        sim.start()
        sim.join()

        self.assertTrue(os.access(file_name, os.F_OK))

        with zipfile.ZipFile(file_name, 'r') as zf:
            self.assertListEqual(zf.namelist(), os.listdir(folder_name))
Example #5
0
    def test_supplemental_output_created(self):
        """
            Ensures that prepare_supplemental_output_directory
            is being called, and that the directory created is
            being passed to adsm properly
        """
        settings = OutputSettings.objects.first()
        settings.save_daily_unit_states = True
        settings.save()
        close_old_connections()
        output_file = os.path.join(self.scenario_directory, 'states_1.csv')

        sim = Simulation(1, testing=True)
        sim.start()
        sim.join()

        self.assertTrue(os.access(output_file, os.F_OK))
Example #6
0
    def test_single_thread(self):
        sim = Simulation(1, testing=True)
        sim.start()
        sim.join()

        self.assertEqual(DailyControls.objects.count(), 10)
Example #7
0
    def test_single_thread(self):
        sim = Simulation(1)
        sim.start()
        sim.join()

        self.assertEqual(DailyReport.objects.count(), 4)