def test_volumetry_evaluation_yaml_generator(self):
"""
Test creates two directories and some empty files. Based on this
structure a yaml file data are constructed.
"""
import os
import shutil
# Creating directory structure
sliver_dir = '__test_sliver_dir'
pklz_dir = '__test_pklz_dir'
# d = os.path.dirname(sliver_dir)
d = sliver_dir
if os.path.exists(d):
shutil.rmtree(d)
# if not os.path.exists(d):
os.makedirs(d)
# d = os.path.dirname(pklz_dir)
d = pklz_dir
if os.path.exists(d):
shutil.rmtree(d)
# if not os.path.exists(d):
os.makedirs(d)
filelist1 = [
'liver-seg001.mhd', 'liver-seg002.mhd', 'liver-seg006.mhd'
]
for fl in filelist1:
open(os.path.join(sliver_dir, fl), 'a').close()
filelist2 = [
'soubor_seg001to.pklz', 'soubor_seg002.pklz', 'sao_seg003.pklz',
'so_seg002tre3.pklz', 'ijij.pklz'
]
for fl in filelist2:
open(os.path.join(pklz_dir, fl), 'a').close()
# construct yaml data
yamldata = ve.generate_input_yaml(sliver_dir, pklz_dir)
# assert
fls0 = os.path.join(sliver_dir, filelist1[0])
flp0 = os.path.join(pklz_dir, filelist2[0])
# we only hope, that this will be first record
self.assertEqual(yamldata['data'][0]['sliverseg'], fls0)
self.assertEqual(yamldata['data'][0]['ourseg'], flp0)
# Clean
# if os.path.exists(d):
shutil.rmtree(sliver_dir)
# if os.path.exists(d):
shutil.rmtree(pklz_dir)
def test_volumetry_evaluation_yaml_generator(self):
"""
Test creates two directories and some empty files. Based on this
structure a yaml file data are constructed.
"""
import os
import shutil
# Creating directory structure
sliver_dir = '__test_sliver_dir'
pklz_dir = '__test_pklz_dir'
# d = os.path.dirname(sliver_dir)
d = sliver_dir
if os.path.exists(d):
shutil.rmtree(d)
# if not os.path.exists(d):
os.makedirs(d)
# d = os.path.dirname(pklz_dir)
d = pklz_dir
if os.path.exists(d):
shutil.rmtree(d)
# if not os.path.exists(d):
os.makedirs(d)
filelist1 = ['liver-seg001.mhd', 'liver-seg002.mhd',
'liver-seg006.mhd']
for fl in filelist1:
open(os.path.join(sliver_dir, fl), 'a').close()
filelist2 = ['soubor_seg001to.pklz', 'soubor_seg002.pklz',
'sao_seg003.pklz', 'so_seg002tre3.pklz', 'ijij.pklz']
for fl in filelist2:
open(os.path.join(pklz_dir, fl), 'a').close()
# construct yaml data
yamldata = ve.generate_input_yaml(
sliver_dir, pklz_dir)
# assert
fls0 = os.path.join(sliver_dir, filelist1[0])
flp0 = os.path.join(pklz_dir, filelist2[0])
# we only hope, that this will be first record
self.assertEqual(yamldata['data'][0]['sliverseg'], fls0)
self.assertEqual(yamldata['data'][0]['ourseg'], flp0)
# Clean
# if os.path.exists(d):
shutil.rmtree(sliver_dir)
# if os.path.exists(d):
shutil.rmtree(pklz_dir)