예제 #1
0
    def test_serialization(self):
        with tempfile.TemporaryDirectory() as tmp_dir:
            path = Path(tmp_dir, "request")
            request = Request(path,
                              "foo",
                              mo.get_global_settings(),
                              save_on_creation=False,
                              enable_logging=False)

            request.default_run.charge = 1
            request.default_run.fluence = 2
            request.default_target.description = "foo"

            request.to_file()

            request_from_file = Request.from_file(request.request_file,
                                                  mo.get_global_settings())

            self.assertEqual(request.default_run.get_settings(),
                             request_from_file.default_run.get_settings())

            self.assertEqual(request.default_target.description,
                             request_from_file.default_target.description)

            utils.disable_logging()
예제 #2
0
    def test_folder_structure(self):
        """Tests the folder structure when request is created"""
        with tempfile.TemporaryDirectory() as tmp_dir:
            path = Path(tmp_dir, self.folder_name)
            Request(path,
                    "bar",
                    mo.get_global_settings(),
                    save_on_creation=False,
                    enable_logging=False)

            self.assertEqual([], os.listdir(tmp_dir))

            request = Request(path,
                              "bar",
                              mo.get_global_settings(),
                              save_on_creation=True,
                              enable_logging=False)

            utils.assert_folder_structure_equal(self.folder_structure, path)

            request.to_file()
            utils.assert_folder_structure_equal(self.folder_structure, path)

            utils.disable_logging()