fex tests

UDST · Apr 4, 2018 · a5b3838 · a5b3838
1 parent bbe9501
commit a5b3838
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/urbansim/models/tests/test_dcm.py b/urbansim/models/tests/test_dcm.py
@@ -219,7 +219,8 @@ def test_mnl_dcm_yaml(basic_dcm, choosers, alternatives):
         'choice_column': basic_dcm.choice_column,
         'fitted': False,
         'log_likelihoods': None,
-        'fit_parameters': None
+        'fit_parameters': None,
+        'normalize': False,
     }
 
     assert yaml.load(basic_dcm.to_yaml()) == expected_dict
@@ -512,14 +513,16 @@ def test_mnl_dcm_segmented_yaml(grouped_choosers, alternatives):
                 'name': 'x',
                 'fitted': False,
                 'log_likelihoods': None,
-                'fit_parameters': None
+                'fit_parameters': None,
+                'normalize': False,
             },
             'y': {
                 'name': 'y',
                 'model_expression': 'var3 + var1:var2',
                 'fitted': False,
                 'log_likelihoods': None,
-                'fit_parameters': None
+                'fit_parameters': None,
+                'normalize': False,
             }
         }
     }