some documentation and tests added for the gui new features

Author: fsalmoir

pygem/gui.py

@@ -38,6 +38,8 @@ class Gui(object):
 	:cvar string url: url of the github page of PyGeM.
 	:cvar Tkinter.Canvas logo_panel: canvas for PyGeM logo.
 	:cvar Tkinter.PhotoImage img: PyGeM logo.
+	:cvar Tkinter.Frame orig_geo_frame: frame for plotting of the original geometry.
+	:cvar Tkinter.Frame mod_geo_frame: frame for plotting of the final geometry.
 	
 	"""
 
@@ -63,8 +65,6 @@ def __init__(self):
 		self.url = 'https://github.com/mathLab/PyGeM'
 		self.logo_panel = None
 		self.img = None
-		
-		
 		self.orig_geo_frame = None
 		self.mod_geo_frame = None
 

pygem/vtkhandler.py

@@ -106,6 +106,9 @@ def plot(self, plot_file=None, save_fig=False):
 		:param string plot_file: the vtk filename you want to plot.
 		:param bool save_fig: a flag to save the figure in png or not. If True the
 			plot is not shown.
+			
+		:return: figure: matlplotlib structure for the figure of the chosen geometry
+		:rtype: matplotlib.pyplot.figure
 		"""
 		if plot_file is None:
 			plot_file = self.infile

tests/test_gui.py

@@ -80,6 +80,16 @@ def test_gui_init_none_4(self):
 		gui_handler = gui.Gui()
 		assert gui_handler.img == None
 
+		
+	def test_gui_init_none_5(self):
+		gui_handler = gui.Gui()
+		assert gui_handler.orig_geo_frame == None
+		
+		
+	def test_gui_init_none_6(self):
+		gui_handler = gui.Gui()
+		assert gui_handler.mod_geo_frame == None
+		
 
 	def test_gui_init_all(self):
 		gui.Gui()