From 61b1d50e74deff96d8cdf83c7b20ad564735190f Mon Sep 17 00:00:00 2001 From: Sam Maurer Date: Wed, 18 Mar 2020 14:33:35 -0700 Subject: [PATCH] Fixing column list comparison --- urbansim/urbanchoice/tests/test_interaction.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/urbansim/urbanchoice/tests/test_interaction.py b/urbansim/urbanchoice/tests/test_interaction.py index 938edcf4..aeca68a5 100644 --- 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)))