test_plot.py 文件源码

def test_let_plot(self):
        working_dir = tempfile.mkdtemp()  # make temp working dir for output file
        output_file = os.path.join(working_dir, "foo.png")

        pytrip.utils.dvhplot.main(args=[self.let, self.vdx, 'target', '-l', '-v', '-o', output_file])

        logger.info("Checking if " + output_file + " is PNG")
        self.assertEqual(imghdr.what(output_file), 'png')

        logger.info("Removing " + working_dir)
        shutil.rmtree(working_dir)