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")