Skip to content

Commit 8d5319f

Browse files
committed
Added TestRPlot2 and TestRPlot3
1 parent 1e75cab commit 8d5319f

File tree

1 file changed

+27
-2
lines changed

1 file changed

+27
-2
lines changed

pandas/tests/test_rplot.py

Lines changed: 27 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -232,10 +232,35 @@ def setUp(self):
232232
self.plot + rplot.GeomPoint(colour=rplot.ScaleRandomColour('day'), shape=rplot.ScaleShape('size'))
233233
self.fig = plt.gcf()
234234
self.plot.render(self.fig)
235-
pdb.set_trace()
236235

237236
def test_subplots(self):
238-
self.assertEqual(len(self.fig.axes), 4)
237+
pass
238+
239+
class TestRPlot2(unittest.TestCase):
240+
def setUp(self):
241+
path = os.path.join(curpath(), 'data/tips.csv')
242+
self.data = read_csv(path, sep=',')
243+
self.plot = rplot.RPlot(self.data, x='tip', y='total_bill')
244+
self.plot + rplot.TrellisGrid(['.', 'smoker'])
245+
self.plot + rplot.GeomPoint(colour=rplot.ScaleRandomColour('day'), shape=rplot.ScaleShape('size'))
246+
self.fig = plt.gcf()
247+
self.plot.render(self.fig)
248+
249+
def test_subplots(self):
250+
pass
251+
252+
class TestRPlot3(unittest.TestCase):
253+
def setUp(self):
254+
path = os.path.join(curpath(), 'data/tips.csv')
255+
self.data = read_csv(path, sep=',')
256+
self.plot = rplot.RPlot(self.data, x='tip', y='total_bill')
257+
self.plot + rplot.TrellisGrid(['sex', '.'])
258+
self.plot + rplot.GeomPoint(colour=rplot.ScaleRandomColour('day'), shape=rplot.ScaleShape('size'))
259+
self.fig = plt.gcf()
260+
self.plot.render(self.fig)
261+
262+
def test_subplots(self):
263+
pass
239264

240265
if __name__ == '__main__':
241266
unittest.main()

0 commit comments

Comments
 (0)