Added the iris data set example to tests

Author: orbitfold

pandas/tests/test_rplot.py

@@ -254,5 +254,16 @@ def test_rplot3(self):
 		self.fig = plt.gcf()
 		self.plot.render(self.fig)
 
+	def test_rplot_iris(self):
+		path = os.path.join(curpath(), 'data/iris.csv')
+		plt.figure()
+		self.data = read_csv(path, sep=',')
+		plot = rplot.RPlot(self.data, x='SepalLength', y='SepalWidth')
+		plot + rplot.GeomPoint(colour=rplot.ScaleGradient('PetalLength', colour1=(0.0, 1.0, 0.5), colour2=(1.0, 0.0, 0.5)), 
+			size=rplot.ScaleSize('PetalWidth', min_size=10.0, max_size=200.0),
+			shape=rplot.ScaleShape('Name'))
+		self.fig = plt.gcf()
+		plot.render(self.fig)
+
 if __name__ == '__main__':
 	unittest.main()

pandas/tools/rplot.py

@@ -138,9 +138,9 @@ def __call__(self, data, index):
 		data: pandas DataFrame
 		index: pandas DataFrame row index
 		"""
-		x = data[column].iget(index)
-		a = float(min(data[column]))
-		b = float(max(data[column]))
+		x = data[self.column].iget(index)
+		a = float(min(data[self.column]))
+		b = float(max(data[self.column]))
 		return self.transform(self.min_size + ((x - a) / (b - a)) * 
 			(self.max_size - self.min_size))