Ejemplo n.º 1
0
    def test_patching(self):
        import astroNN.data
        from astroNN.shared.patch_util import Patch

        diff = os.path.join(astroNN.data.datapath(), 'tf1_12.patch')
        patch = Patch(diff)
        patch_file_path = "travis_tf_1_12.py"
        if os.path.exists(patch_file_path) is False:
            patch_file_path = os.path.join("tests", "travis_tf_1_12.py")

        with open(patch_file_path, 'r') as f:
            original_text = f.read()

        patch.apply(patch_file_path)
        with open(patch_file_path, 'r') as f:
            patched_text = f.read()

        patch.apply(patch_file_path)
        with open(patch_file_path, 'r') as f:
            patched_twice_text = f.read()

        patch.revert(patch_file_path)
        with open(patch_file_path, 'r') as f:
            unpatched_text = f.read()

        patch.revert(patch_file_path)
        with open(patch_file_path, 'r') as f:
            unpatched_twice_text = f.read()

        # assert patching, patching twice and unpatching work correctly
        self.assertNotEqual(original_text, patched_text)
        self.assertEqual(patched_twice_text, patched_text)
        self.assertEqual(original_text, unpatched_text)
        self.assertEqual(unpatched_twice_text, unpatched_text)
Ejemplo n.º 2
0
    def __master_patch(path, diffpatch):
        # generate patch and patch
        patch = Patch(diffpatch)
        if method == 'patch':
            action_name = 'Patched'
            err = patch.apply(path)
        elif method == 'unpatch':
            action_name = 'Unpatched'
            err = patch.revert(path)
        else:
            raise ValueError(f"Unknown method={method}")

        if err == 0:
            print(f"{action_name} Successfully at {path}!")
        else:
            print("Error Occurred!")