adds test for add_bars

Brian Luft · Brian Luft · commit affb9f197acc · 2011-02-20T18:08:00.000-08:00
diff --git a/pyflot/__init__.py b/pyflot/__init__.py
@@ -1,5 +1,4 @@
 import collections
-
 from functools import partial
 from itertools import chain
 import inspect
@@ -58,7 +57,6 @@ def __init__(self):
             if hasattr(base, 'options'):
                 update(self._options, base.options)
 
-
     @property
     def series_json(self):
         """
@@ -77,10 +75,14 @@ def options_json(self):
         """
         return json.dumps(self._options)
 
-    #add_bars
-    #add_line
-    #add_points
     def __getattr__(self, value):
+        """
+        add_bars
+        add_line
+        add_points
+        
+        provides shortcut methods for adding series using a particular line type
+        """
         if value.startswith('add_'):
             return partial(self.add_series_type, value[4:])
 
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -99,6 +99,14 @@ def test_add_time_series(self):
         self.assertEqual(self.flot._options['xaxis'],
                 {'mode': 'time'})
 
+    def test_add_bars(self):
+        "test the shortcut for adding a series as bars"
+        series = {'data': ((0,0), ) + S1, 
+                  "bars": {"barWidth": 0.75, "show": True}}
+        self.flot.add_bars(((0,0), ) + S1)
+        self.assertEqual(self.flot._series[0]['bars'], {'show': True})
+        self.assertEqual(json.dumps([series]), self.flot.series_json)
+
     def test_empty_options_json(self):
         "make sure conversion to JSON works for default options"
         self.assertEqual("{}", self.flot.options_json)
