BUG: fix ujson handling of new series object

Author: Komnomnomnom

pandas/io/tests/test_json/test_ujson.py

@@ -1150,7 +1150,6 @@ def testSeries(self):
         s = Series([10, 20, 30, 40, 50, 60], name="series", index=[6,7,8,9,10,15])
         s.sort()
 
-        import pdb; pdb.set_trace()
         # column indexed
         outp = Series(ujson.decode(ujson.encode(s)))
         outp.sort()

pandas/src/ujson/python/objToJSON.c

@@ -1387,11 +1387,13 @@ void Object_beginTypeContext (JSOBJ _obj, JSONTypeContext *tc)
       return;
     }
 
+    pc->newObj = PyObject_GetAttrString(obj, "values");
+
     if (enc->outputFormat == INDEX || enc->outputFormat == COLUMNS)
     {
       PRINTMARK();
       tc->type = JT_OBJECT;
-      pc->columnLabelsLen = PyArray_SIZE(obj);
+      pc->columnLabelsLen = PyArray_DIM(pc->newObj, 0);
       pc->columnLabels = NpyArr_encodeLabels((PyArrayObject*) PyObject_GetAttrString(obj, "index"), (JSONObjectEncoder*) enc, pc->columnLabelsLen);
       if (!pc->columnLabels)
       {
@@ -1403,7 +1405,6 @@ void Object_beginTypeContext (JSOBJ _obj, JSONTypeContext *tc)
       PRINTMARK();
       tc->type = JT_ARRAY;
     }
-    pc->newObj = PyObject_GetAttrString(obj, "values");
     pc->iterBegin = NpyArr_iterBegin;
     pc->iterEnd = NpyArr_iterEnd;
     pc->iterNext = NpyArr_iterNext;