test_copyto.py

# coding: utf-8
# Copyright (c) Max-Planck-Institut für Eisenforschung GmbH - Computational Materials Design (CM) Department
# Distributed under the terms of "New BSD License", see the LICENSE file.

import os
from pyiron_base.project.generic import Project
from pyiron_base._tests import TestWithProject

class TestCopyTo(TestWithProject):

    def test_copy_to_job(self):
        job_ser = self.project.create.job.SerialMasterBase("sequence_single")
        ham = self.project.create.job.ScriptJob("job_single")
        ham.copy_to(job_ser)
        self.assertTrue(job_ser['job_single/input/custom_dict'] is not None)
        job_ser.remove()

        job = self.project.create.job.ScriptJob("job_script")

        with open('foo.py', 'w') as f:
            f.write("print(42)\n")

        job.script_path = "foo.py"
        job.save()
        copy = job.copy_to(job.project_hdf5, "job_copy")
        self.assertEqual(job.script_path, copy.script_path,
                         "Script path not equal after copy.")

        jobc = self.project.inspect(job.id)
        copyc = jobc.copy_to(jobc.project_hdf5, "job_core_copy")
        self.assertEqual(jobc.to_object().script_path, copyc.to_object().script_path,
                         "Script path not equal after JobCore copy.")

        os.remove("foo.py")

    def test_copy_to_project(self):
        sub_project = self.project.copy()
        sub_project = sub_project.open("sub_project")
        ham = self.project.create_job("ScriptJob", "job_single_pr")
        ham.copy_to(project=sub_project)

    def test_copy_to_job_ex(self):
        job_ser = self.project.create.job.SerialMasterBase("sequence_single_ex")
        ham = self.project.create.job.ScriptJob("job_single_ex")
        ham.to_hdf()
        ham.copy_to(job_ser)
        self.assertTrue(job_ser['job_single_ex/input/custom_dict'] is not None,
                        "Input not copied!")
        ham.remove()
        job_ser.remove()

    def test_copy_to_project_ex(self):
        sub_project = self.project.copy()
        sub_project = sub_project.open("sub_project_ex")
        ham = self.project.create_job("ScriptJob", "job_single_pr_ex")
        ham.to_hdf()
        ham.copy_to(project=sub_project)
        ham.remove()
        os.remove(
            os.path.join(
                self.file_location, f"{sub_project.path}/job_single_pr_ex.h5"
            )
        )


if __name__ == "__main__":
    unittest.main()