fix MultiIndex initiation from np.nan

Author: OXPHOS

pandas/tools/tests/test_pivot.py

@@ -1094,19 +1094,22 @@ def test_margin_dropna(self):
 
         actual = pd.crosstab(a, [b, c], rownames=['a'],
                              colnames=['b', 'c'], margins=True, dropna=False)
-        m = MultiIndex.from_arrays([[np.nan, np.nan, 'one', 'one', 'two', 'two', 'All'],
-                                    ['dull', 'shiny', 'dull', 'shiny', 'dull', 'shiny', '']],
-                                   names=['b', 'c'])
+
+        m = MultiIndex(levels = [Index(['All', np.nan, 'one', 'two']),
+                        Index(['', 'dull', 'shiny'])], labels = [[1, 1, 2, 2, 3, 3, 0],
+                        [1, 2, 1, 2, 1, 2, 0]], names=['b', 'c'])
         expected = DataFrame([[0, 0, 1, 0, 1, 0, 2], [0, 1, 2, 0, 1, 1, 5],
                               [0, 1, 3, 0, 2, 1, 7]], columns=m)
         expected.index = Index(['bar', 'foo', 'All'], name='a')
         tm.assert_frame_equal(actual, expected)
 
         actual = pd.crosstab([a, b], c, rownames=['a', 'b'],
                              colnames=['c'], margins=True, dropna=False)
-        m = MultiIndex.from_arrays([['bar', 'bar', 'bar', 'foo', 'foo', 'foo', 'All'],
-                                    [np.nan, 'one', 'two', np.nan, 'one', 'two', '']],
-                                   names=['a', 'b'])
+
+        print actual.index
+        m = MultiIndex(levels=[['All', 'bar', 'foo'], ['', np.nan, 'one', 'two']],
+                        labels=[[1, 1, 1, 2, 2, 2, 0], [1, 2, 3, 1, 2, 3, 0]],
+                        names=['a', 'b'])
         expected = DataFrame([[0, 0, 0], [1, 0, 1], [1, 0, 1], [0, 1, 1], [2, 0, 2], [1, 1, 2],
                               [5, 2, 7]], index=m)
         expected.columns = Index(['dull', 'shiny', 'All'], name='c')