more tests for deterministic names

mrocklin · Aug 4, 2015 · 9ad488e · 9ad488e
1 parent e388ed8
commit 9ad488e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/dask/dataframe/tests/test_dataframe.py b/dask/dataframe/tests/test_dataframe.py
@@ -494,6 +494,8 @@ def test_loc():
     assert eq(d.loc[1000:], full.loc[1000:])
     assert eq(d.loc[-2000:-1000], full.loc[-2000:-1000])
 
+    assert sorted(d.loc[5].dask) == sorted(d.loc[5].dask)
+    assert sorted(d.loc[5].dask) != sorted(d.loc[6].dask)
 
 
 def test_loc_with_text_dates():
@@ -510,6 +512,9 @@ def test_loc_with_text_dates():
 def test_loc_with_series():
     assert eq(d.loc[d.a % 2 == 0], full.loc[full.a % 2 == 0])
 
+    assert sorted(d.loc[d.a % 2].dask) == sorted(d.loc[d.a % 2].dask)
+    assert sorted(d.loc[d.a % 2].dask) != sorted(d.loc[d.a % 3].dask)
+
 
 def test_iloc_raises():
     assert raises(AttributeError, lambda: d.iloc[:5])
@@ -781,6 +786,8 @@ def test_deterministic_arithmetic_names():
     a = dd.from_pandas(df, npartitions=2)
 
     assert sorted((a.x + a.y ** 2).dask) == sorted((a.x + a.y ** 2).dask)
+    assert sorted((a.x + a.y ** 2).dask) != sorted((a.x + a.y ** 3).dask)
+    assert sorted((a.x + a.y ** 2).dask) != sorted((a.x - a.y ** 2).dask)
 
 
 def test_deterministic_reduction_names():