Fixing column list comparison

UDST · Mar 18, 2020 · 61b1d50 · 61b1d50
1 parent 9c43d04
commit 61b1d50
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/urbansim/urbanchoice/tests/test_interaction.py b/urbansim/urbanchoice/tests/test_interaction.py
@@ -35,8 +35,8 @@ def test_interaction_dataset_sim(choosers, alternatives):
 
     assert len(merged) == len(choosers) * len(alternatives)
     npt.assert_array_equal(merged.index.values, sample)
-    assert list(merged.columns) == [
-        'var2', 'var3', 'join_index', 'thing_id', 'var1']
+    assert set(list(merged.columns)) == set([
+        'var2', 'var3', 'join_index', 'thing_id', 'var1'])
     npt.assert_array_equal(
         merged['var1'].values,
         choosers['var1'].values.repeat(len(alternatives)))