Remove unit matching in flow mapping-to-flowlist test to allow altern…

…ate units
USEPA · Dec 12, 2019 · c0bec27 · c0bec27
1 parent e648f14
commit c0bec27
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_flow_mappings.py b/tests/test_flow_mappings.py
@@ -36,8 +36,8 @@ def test_targetflowinfo_matches_flows_in_list(self):
         """Checks for duplicate uuids in list
         """
         flowmapping_targetinfo = self.flowmappings[['TargetFlowName', 'TargetFlowUUID',
-                                                    'TargetFlowContext', 'TargetUnit']]
-        flowmapping_targetinfo.columns = ['Flowable', 'Flow UUID', 'Context', 'Unit']
+                                                    'TargetFlowContext']]
+        flowmapping_targetinfo.columns = ['Flowable', 'Flow UUID', 'Context']
         flowmappings_w_flowlist = pd.merge(flowmapping_targetinfo,self.flowlist)
         # To identify flowmapping flows not in list
         missing_flows = flowmapping_targetinfo[~flowmapping_targetinfo['Flow UUID'].isin(flowmappings_w_flowlist['Flow UUID'])]