TST: drop unneeded demographics field

jeffgortmaker · May 13, 2018 · 3a0a9e9 · 3a0a9e9
1 parent ed48ab1
commit 3a0a9e9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -188,7 +188,8 @@ def knittel_metaxoglou_2014():
             'characteristics': characteristics
         })]
     })
-    problem = Problem(product_data, np.recfromcsv(BLP_AGENTS_LOCATION))
+    agent_data = np.lib.recfunctions.drop_fields(np.recfromcsv(BLP_AGENTS_LOCATION), 'demographics0')
+    problem = Problem(product_data, agent_data)
     return scipy.io.loadmat(str(TEST_DATA_PATH / 'knittel_metaxoglou_2014.mat'), {'problem': problem})