def test_print_map_start(self):
captured_output = io.StringIO()
sys.stdout = captured_output
sys.argv = [
"something", "-motl", "emd_3465.map", "tests/motl_1_1t.csv",
"--tomogram-map-file", "emd_3464.map", "-o", "rm_c.txt", "-c", "-s"
]
args = cm.parse_args()
cm.create_output_motl(args)
self.assertTrue(
"nxstart: 0\nnystart: 0\nnzstart: 0" in captured_output.getvalue())
os.remove("rm_c.txt")
def test_output_name_specified_c(self):
captured_output = io.StringIO()
sys.stdout = captured_output
sys.argv = [
"something", "-motl", "emd_3465.map", "tests/motl_1_1t.csv",
"--tomogram-map-file", "emd_3464.map", "-o", "rm_c.txt", "-c"
]
args = cm.parse_args()
cm.create_output_motl(args)
self.assertTrue(os.path.exists("rm_c.txt"))
self.assertTrue("is compressed" in captured_output.getvalue())
os.remove("rm_c.txt")
def test_print_voxel_size(self):
captured_output = io.StringIO()
sys.stdout = captured_output
sys.argv = [
"something", "-motl", "emd_3465.map", "tests/motl_1_1t.csv",
"--tomogram-map-file", "emd_3464.map", "-o", "rm_c.txt", "-c",
"-s", "-v"
]
args = cm.parse_args()
cm.create_output_motl(args)
self.assertTrue("voxel_size in x, y, z: (7.12, 7.12, 7.12)" in
captured_output.getvalue())
os.remove("rm_c.txt")
def test_check_output_data(self):
sys.argv = [
"something", "-motl", "emd_3465.map", "tests/motl_1_1t.csv",
"--tomogram-map-file", "emd_3464.map"
]
args = cm.parse_args()
cm.create_output_motl(args)
with open("output_nc.txt",
"r") as text: # output_nc.txt created in the last step
data = text.readlines()
for each in data:
if re.match("^Data", each):
string_first_10 = each.replace("Data:\t", "")[0:10]
self.assertEqual('wLHtOYrHeD', string_first_10)
os.remove("output_nc.txt")
def test_compressed(self):
sys.argv = [
"something", "-motl", "emd_3465.map", "tests/motl_1_1t.csv",
"--tomogram-map-file", "emd_3464.map", "-c"
]
args = cm.parse_args()
cm.create_output_motl(args)
self.assertTrue(os.path.exists("output_c.txt"))
with open("output_c.txt", "r") as text:
data = text.readlines()
for each in data:
if re.match("^Data", each):
string_first_10 = each.replace("Data:\t", "")[0:10]
print(string_first_10)
self.assertEqual("eJwUm/cj1e", string_first_10)
os.remove("output_c.txt")