ensure that index information passes through castra

mrocklin · Jul 15, 2015 · 72bbb7d · 72bbb7d
1 parent b4a1de5
commit 72bbb7d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dask/dataframe/tests/test_io.py b/dask/dataframe/tests/test_io.py
@@ -399,12 +399,15 @@ def test_from_dask_array_raises():
 def test_to_castra():
     pytest.importorskip('castra')
     df = pd.DataFrame({'x': ['a', 'b', 'c', 'd'],
-                       'y': [1, 2, 3, 4]})
+                       'y': [1, 2, 3, 4]},
+                       index=pd.Index([1., 2., 3., 4.], name='ind'))
     a = dd.from_pandas(df, 2)
 
     c = a.to_castra()
+    b = c.to_dask()
     try:
         tm.assert_frame_equal(df, c[:])
+        tm.assert_frame_equal(b.compute(), df)
     finally:
         c.drop()