def test_voltage_labels(self):
    expected_files = ['KP_voltage.png']
    cursor = res2dict((REAL_PATH + 'data/hull-KP-KSnP_pub/*.res'))[0]
    hull = QueryConvexHull(cursor=cursor, species='KP', no_plot=True, voltage=True, labels=True)
    plot_voltage_curve(hull.voltage_data, labels=True, savefig=expected_files[0])
    for expected_file in expected_files:
        self.assertTrue(os.path.isfile(expected_file))
def test_voltage_labels(self):
    expected_files = ['KP_voltage.png']
    cursor = res2dict((REAL_PATH + 'data/hull-KP-KSnP_pub/*.res'))[0]
    hull = QueryConvexHull(cursor=cursor, species='KP', no_plot=True, voltage=True, labels=True)
    plot_voltage_curve(hull.voltage_data, labels=True, savefig=expected_files[0])
    for expected_file in expected_files:
        self.assertTrue(os.path.isfile(expected_file))