Fix issue with undeterministic test case

has2k1 · Jul 31, 2018 · 83183df · 83183df
1 parent 0ab88f2
commit 83183df
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 3 deletions.
diff --git a/plotnine/scales/scale.py b/plotnine/scales/scale.py
@@ -221,7 +221,7 @@ def train_df(self, df):
         """
         Train scale from a dataframe
         """
-        aesthetics = set(self.aesthetics) & set(df.columns)
+        aesthetics = sorted(set(self.aesthetics) & set(df.columns))
         for ae in aesthetics:
             self.train(df[ae])
 

diff --git a/plotnine/tests/baseline_images/test_scale_internals/multiple_aesthetics.png b/plotnine/tests/baseline_images/test_scale_internals/multiple_aesthetics.png
diff --git a/plotnine/tests/test_scale_internals.py b/plotnine/tests/test_scale_internals.py
@@ -402,8 +402,6 @@ def scale_scale_discrete_mapping_nulls():
     assert res[2] == expected[2]
 
 
-@pytest.mark.skipif(
-    six.PY2, reason="I do not know why it fails on travis")
 def test_multiple_aesthetics():
 
     df = pd.DataFrame({