mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
import os
|
|
import shutil
|
|
from unittest import TestCase
|
|
|
|
from dvc.path.path import DvcPath
|
|
from dvc.git_wrapper import GitWrapperI
|
|
|
|
|
|
class TestDvcPathTest(TestCase):
|
|
def setUp(self):
|
|
self.test_dir = os.path.realpath('/tmp/ntx_unit_test/dvc_path')
|
|
shutil.rmtree(self.test_dir, ignore_errors=True)
|
|
os.makedirs(os.path.join(self.test_dir, 'data'))
|
|
os.makedirs(os.path.join(self.test_dir, 'code', 'lib'))
|
|
os.makedirs(os.path.join(self.test_dir, 'd1', 'd2', 'dir3', 'd4', 'dir5'))
|
|
|
|
self._git = GitWrapperI(self.test_dir)
|
|
|
|
def _validate_dvc_path(self, path, dvc_file_name, relative_file_name):
|
|
self.assertEqual(path.dvc, dvc_file_name)
|
|
self.assertEqual(path.filename, os.path.basename(dvc_file_name))
|
|
self.assertEqual(path.relative, relative_file_name)
|
|
self.assertEquals(path.abs[0], os.path.sep)
|
|
self.assertTrue(path.abs.endswith(dvc_file_name))
|
|
|
|
def basic_test(self):
|
|
os.chdir(self.test_dir)
|
|
|
|
file = os.path.join('data', 'file.txt')
|
|
path = DvcPath(file, self._git)
|
|
self._validate_dvc_path(path, file, file)
|
|
pass
|
|
|
|
def from_dir_test(self):
|
|
os.chdir(os.path.join(self.test_dir, 'code'))
|
|
|
|
file_dvc_path = os.path.join('data', 'file1.txt')
|
|
file_relative_path = os.path.join('..', file_dvc_path)
|
|
|
|
path = DvcPath(file_relative_path, self._git)
|
|
self._validate_dvc_path(path, file_dvc_path, file_relative_path)
|
|
pass
|
|
|
|
def from_deep_dirs_test(self):
|
|
deep_dir = os.path.join('d1', 'd2', 'dir3')
|
|
os.chdir(os.path.join(self.test_dir, deep_dir))
|
|
|
|
file_dvc = os.path.join('code', 'lib', 'context_switcher_structs.asm')
|
|
file_relative = os.path.join('..', '..', '..', file_dvc)
|
|
|
|
path = DvcPath(file_relative, self._git)
|
|
self._validate_dvc_path(path, file_dvc, file_relative)
|
|
pass
|
|
|
|
def go_deeper_test(self):
|
|
deep_dir = os.path.join('d1', 'd2', 'dir3')
|
|
os.chdir(os.path.join(self.test_dir, deep_dir))
|
|
|
|
file_relative_path = os.path.join(deep_dir, 'd4', 'dir5', 'rawdata.tsv')
|
|
file_dvc_path = os.path.join(deep_dir, file_relative_path)
|
|
|
|
path = DvcPath(file_relative_path, self._git)
|
|
self._validate_dvc_path(path, file_dvc_path, file_relative_path)
|
|
pass
|