Handle case of traffic_allocation being None

Apostolos Bessas · Apostolos Bessas · commit d6d094135fd0 · 2017-05-25T13:46:58.000+03:00
In Python 3 you cannot compare None with an integer. As a result, the localhost
implementation fails.
diff --git a/splitio/splits.py b/splitio/splits.py
@@ -76,7 +76,9 @@ def __init__(self, name, seed, killed, default_treatment, traffic_type_name,
         self._change_number = change_number
         self._conditions = conditions if conditions is not None else []
 
-        if traffic_allocation >= 0 and traffic_allocation <= 100:
+        if traffic_allocation is None:
+            self._traffic_allocation = 100
+        elif traffic_allocation >= 0 and traffic_allocation <= 100:
             self._traffic_allocation = traffic_allocation
         else:
             self._traffic_allocation = 100
