def test_pdf_plot(self):
    structure = res2dict((REAL_PATH + 'data/res_files/KPSn-OQMD_123456.res'), as_model=True)[0]
    plot_pdf(structure, png=True)
    self.assertTrue(os.path.isfile('K7PSn_pdf.png'))
    plot_pdf([structure, structure], filename='test_pdf', rmax=5, png=True)
    self.assertTrue(os.path.isfile('test_pdf.png'))
def test_pdf_plot(self):
    structure = res2dict((REAL_PATH + 'data/res_files/KPSn-OQMD_123456.res'), as_model=True)[0]
    plot_pdf(structure, png=True)
    self.assertTrue(os.path.isfile('K7PSn_pdf.png'))
    plot_pdf([structure, structure], filename='test_pdf', rmax=5, png=True)
    self.assertTrue(os.path.isfile('test_pdf.png'))