mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-07 15:28:28 +01:00
Ruslan Kuprieiev
f47a2c7585
Initial unittests for new design. Signed-off-by: Ruslan Kuprieiev <[email protected]>
101 lines
4.1 KiB
Python
101 lines
4.1 KiB
Python
import os
|
|
|
|
from tests.basic_env import TestDvc
|
|
|
|
|
|
class ReproBasicEnv(TestDvc):
|
|
def setUp(self):
|
|
super(ReproBasicEnv, self).setUp()
|
|
|
|
os.mkdir('data')
|
|
|
|
self.file_name1 = os.path.join('data', 'file1')
|
|
self.file1_code_file = 'file1.py'
|
|
self.create(self.file1_code_file, 'print("Hello")' + os.linesep + 'print("Mary")')
|
|
stage = self.dvc.run(fname='file1.dvc',
|
|
outs=[self.file_name1],
|
|
deps_no_cache=[self.file1_code_file],
|
|
cmd='python {} --not-repro > {}'.format(self.file1_code_file,
|
|
self.file_name1))
|
|
|
|
self.file_name11 = os.path.join('data', 'file11')
|
|
self.file11_code_file = 'file11.py'
|
|
self.create(self.file11_code_file,
|
|
'import sys' + os.linesep + 'print(open(sys.argv[1]).readline().strip())')
|
|
|
|
stage = self.dvc.run(fname='file11.dvc',
|
|
outs=[self.file_name11],
|
|
deps=[self.file_name1],
|
|
deps_no_cache=[self.file11_code_file],
|
|
cmd='python {} {} > {}'.format(self.file11_code_file,
|
|
self.file_name1,
|
|
self.file_name11))
|
|
self.file_name11_stage_name = stage.path
|
|
|
|
self.file_name2 = os.path.join('data', 'file2')
|
|
self.file2_code_file = 'file2.py'
|
|
self.create(self.file2_code_file, 'print("Bobby")')
|
|
self.dvc.run(fname='file2.dvc',
|
|
outs=[self.file_name2],
|
|
cmd='python {} --not-repro > {}'.format(self.file2_code_file,
|
|
self.file_name2))
|
|
|
|
self.file_res_code_file = 'code_res.py'
|
|
self.create(self.file_res_code_file,
|
|
'import sys' + os.linesep +
|
|
'text1 = open(sys.argv[1]).read()' + os.linesep +
|
|
'text2 = open(sys.argv[2]).read()' + os.linesep +
|
|
'print(text1 + text2)')
|
|
self.file_name_res = os.path.join('data', 'file_res')
|
|
stage = self.dvc.run(fname='file_res.dvc',
|
|
outs=[self.file_name_res],
|
|
deps_no_cache=[self.file_res_code_file],
|
|
deps=[self.file_name11, self.file_name2],
|
|
cmd='python {} {} {} > {}'.format(self.file_res_code_file,
|
|
self.file_name11,
|
|
self.file_name2,
|
|
self.file_name_res))
|
|
self.file_name_res_stage_name = stage.path
|
|
|
|
def modify_file(self, filename, content_to_add=' '):
|
|
fd = open(filename, 'a')
|
|
fd.write(content_to_add)
|
|
fd.close()
|
|
|
|
|
|
class ReproCodeDependencyTest(ReproBasicEnv):
|
|
def test(self):
|
|
self.modify_file(self.file_res_code_file)
|
|
|
|
self.dvc.reproduce([self.file_name_res_stage_name])
|
|
|
|
self.assertEqual(open(self.file_name_res).read().strip(), 'Hello\nBobby')
|
|
|
|
|
|
class ReproChangedDependency(ReproBasicEnv):
|
|
def test(self):
|
|
self.recreate_file1()
|
|
|
|
self.dvc.reproduce([self.file_name11_stage_name])
|
|
|
|
self.assertEqual(open(self.file_name11).read(), 'Goodbye\n')
|
|
|
|
def recreate_file1(self):
|
|
self.dvc.remove(self.file_name1)
|
|
|
|
file1_code_file = 'file1_2.py'
|
|
self.create(file1_code_file, 'print("Goodbye")' + os.linesep + 'print("Jack")')
|
|
self.dvc.run(fname='file1.dvc',
|
|
outs=[self.file_name1],
|
|
deps_no_cache=[file1_code_file],
|
|
cmd='python {} --not-repro > {}'.format(file1_code_file, self.file_name1))
|
|
|
|
|
|
class ReproChangedDeepDependency(ReproChangedDependency):
|
|
def test(self):
|
|
self.recreate_file1()
|
|
|
|
self.dvc.reproduce([self.file_name_res_stage_name])
|
|
|
|
self.assertEqual(open(self.file_name_res).read().strip(), 'Goodbye\nBobby')
|