tests should pass on pandas<0.24

KrishnaswamyLab · Aug 9, 2019 · c71292a · c71292a
1 parent 3895722
commit c71292a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/test_data.py b/test/test_data.py
@@ -69,7 +69,11 @@ def test_pandas_dataframe():
 
 
 def test_pandas_sparse_dataframe():
-    G = build_graph(pd.DataFrame(data).astype(pd.SparseDtype(float, fill_value=0)))
+    try:
+        X = pd.DataFrame(data).astype(pd.SparseDtype(float, fill_value=0))
+    except AttributeError:
+        X = pd.SparseDataFrame(data, default_fill_value=0)
+    G = build_graph(X)
     assert isinstance(G, graphtools.base.BaseGraph)
     assert isinstance(G.data, sp.csr_matrix)