test_import.py 文件源码

def test_import_file_to_dir(self):
        with tempfile.NamedTemporaryFile(mode='w') as temp:
            content = 'Some data'
            temp.write(content)
            temp.flush()

            dir = os.path.join('data', self.dir1)
            settings = copy(self.settings)
            settings.parse_args('import {} {}'.format(temp.name, dir))
            cmd = CmdImportFile(settings)
            cmd.run()

            dvc_name = os.path.join(dir, os.path.basename(temp.name))
            data_item = self.path_factory.existing_data_item(dvc_name)

            self.assertTrue(os.path.exists(data_item.data.relative))
            self.assertTrue(filecmp.cmp(data_item.data.relative, data_item.cache.relative))
            self.assertEqual(open(data_item.data.relative).read(), content)

            self.assertTrue(os.path.exists(data_item.cache.relative))
            self.assertEqual(open(data_item.cache.relative).read(), content)

            self.assertTrue(os.path.exists(data_item.state.relative))
        pass