Adds axis label setters

Author: palnabarun

mplaltair/_axis.py

@@ -1,10 +1,28 @@
 import numpy as np
 from matplotlib import pyplot as plt
 
-def _handle_x(ax, spec):
+def _set_title(ax, enc, spec):
+    title = spec.get('title')
+    aggregate = spec.get('aggregate')
+    bin = spec.get('bin')
+    timeUnit = spec.get('timeUnit')
+
+    if title:
+        if enc == 'x':
+            ax.set_xlabel(title)
+        elif enc == 'y':
+            ax.set_ylabel(title)
+    elif aggregate:
+        raise NotImplementedError
+    elif bin:
+        raise NotImplementedError
+    elif timeUnit:
+        raise NotImplementedError
+
+def _set_ticks(ax, enc, spec):
     pass
 
-def _handle_y(ax, spec):
+def _set_locators(ax, enc, spec):
     pass
 
 def convert_axis(ax, chart):
@@ -22,11 +40,8 @@ def convert_axis(ax, chart):
 
     """
 
-    _AXIS_HANDLER_MAPPING = {
-        'x': _handle_x,
-        'y': _handle_y,
-    }
-
     for enc_channel, enc_spec in chart.to_dict['encoding']:
         if enc_channel in ['x', 'y']:
-            _AXIS_HANDLER_MAPPING[enc_channel](ax, enc_spec)
+            _set_title(ax, enc_channel, enc_spec)
+            _set_ticks(ax, enc_channel, enc_spec)
+            _set_locators(ax, enc_channel, enc_spec)