Merge pull request #1143 from fedarko/master

Update test_predicates to include field predicates
vega · Sep 22, 2018 · 2bca553 · 2bca553
2 parents d71af34 + b09bf6d
commit 2bca553
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/altair/vegalite/v2/api.py b/altair/vegalite/v2/api.py
@@ -333,7 +333,11 @@ def condition(predicate, if_true, if_false, **kwargs):
                             core.SelectionAnd, core.SelectionOperand)
     test_predicates = (six.string_types, expr.Expression, core.Predicate,
                        core.LogicalOperandPredicate, core.LogicalNotPredicate,
-                       core.LogicalOrPredicate, core.LogicalAndPredicate)
+                       core.LogicalOrPredicate, core.LogicalAndPredicate,
+                       core.FieldEqualPredicate, core.FieldOneOfPredicate,
+                       core.FieldRangePredicate, core.FieldLTPredicate,
+                       core.FieldGTPredicate, core.FieldLTEPredicate,
+                       core.FieldGTEPredicate, core.SelectionPredicate)
 
     if isinstance(predicate, NamedSelection):
         condition = {'selection': predicate._get_name()}