Fix tests for previous changes

brightway-lca · Jul 23, 2024 · 6d4ae78 · 6d4ae78
1 parent c4faf28
commit 6d4ae78
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 3 deletions.
diff --git a/tests/integration/test_brightway_export_allocation.py b/tests/integration/test_brightway_export_allocation.py
@@ -47,13 +47,23 @@ def test_basic_header_extraction(fixtures_dir):
         "comment": "INDIA",
         "functional": True,
         "line_no": 211,
+        "mf_allocated": True,
+        "mf_manual_input_product": False,
         "name": "Rice, at farm (WFLDB 3.0)/IN U",
         "properties": {"manual_allocation": 95.8},
         "type": "production",
         "unit": "kg",
         "waste_type": "not defined",
     }
-    assert allocated["exchanges"][0] == expected
+    for key, value in expected.items():
+        assert allocated["exchanges"][0][key] == value
+
+    assert allocated["exchanges"][0]["input"]
+    assert allocated["exchanges"][0]["mf_allocated_process_code"]
+    assert (
+        allocated["exchanges"][0]["input"][1]
+        == allocated["exchanges"][0]["mf_allocated_process_code"]
+    )
 
     expected = {
         "allocation": 4.2,
@@ -66,12 +76,22 @@ def test_basic_header_extraction(fixtures_dir):
         "assumption that grains account for "
         "92% and straw for 8% of the price.",
         "functional": True,
+        "mf_allocated": True,
+        "mf_manual_input_product": False,
         "line_no": 212,
         "name": "Rice straw, at farm (WFLDB 3.0)/IN U",
         "properties": {"manual_allocation": 4.2},
         "type": "production",
         "unit": "kg",
         "waste_type": "not defined",
     }
-    assert allocated["exchanges"][1] == expected
-    assert len(allocated["exchanges"])
+
+    for key, value in expected.items():
+        assert allocated["exchanges"][1][key] == value
+
+    assert allocated["exchanges"][1]["input"]
+    assert allocated["exchanges"][1]["mf_allocated_process_code"]
+    assert (
+        allocated["exchanges"][1]["input"][1]
+        == allocated["exchanges"][1]["mf_allocated_process_code"]
+    )
diff --git a/tests/integration/test_process_csv.py b/tests/integration/test_process_csv.py
@@ -18,6 +18,7 @@ def test_basic_header_extraction(fixtures_dir):
         "decimal_separator": ",",
         "delimiter": ";",
         "export_platform_ids": None,
+        'exclude_library_processes': None,
         "include_stages": None,
         "kind": SimaProCSVType.processes,
         "libraries": [],

diff --git a/tests/test_header.py b/tests/test_header.py
@@ -24,6 +24,7 @@ def test_basic_header_extraction(fixtures_dir):
         "dayfirst": False,
         "selection": "Selection (2)",
         "open_project": "Bobs_burgers",
+        "exclude_library_processes": None,
         "open_library": None,
         "date_separator": "/",
         "export_platform_ids": None,
@@ -46,6 +47,7 @@ def test_basic_header_extraction(fixtures_dir):
         "dayfirst": True,
         "selection": "Current (1)",
         "open_project": None,
+        "exclude_library_processes": None,
         "open_library": None,
         "date_separator": "/",
         "export_platform_ids": None,
@@ -68,6 +70,7 @@ def test_basic_header_extraction(fixtures_dir):
         "dayfirst": True,
         "selection": "Current (1)",
         "open_project": "Dummy project",
+        "exclude_library_processes": None,
         "open_library": None,
         "date_separator": "/",
         "export_platform_ids": False,
@@ -90,6 +93,7 @@ def test_basic_header_extraction(fixtures_dir):
         "libraries": [],
         "dayfirst": False,
         "selection": "Selection (1)",
+        "exclude_library_processes": None,
         "open_project": None,
         "open_library": "Methods",
         "date_separator": "-",
@@ -111,6 +115,7 @@ def test_basic_header_extraction(fixtures_dir):
         "csv_version": "9.0.0",
         "libraries": [],
         "dayfirst": True,
+        "exclude_library_processes": None,
         "selection": None,
         "open_project": None,
         "open_library": None,
@@ -161,6 +166,7 @@ def test_basic_header_extraction(fixtures_dir):
         "dayfirst": False,
         "selection": "Selection (1)",
         "open_project": "AP9 pilot",
+        "exclude_library_processes": None,
         "open_library": None,
         "date_separator": "/",
         "export_platform_ids": True,
@@ -192,6 +198,7 @@ def test_minimal_header_extraction(fixtures_dir):
         "dayfirst": False,
         "date_separator": "/",
         "open_project": None,
+        "exclude_library_processes": None,
         "open_library": None,
         "export_platform_ids": None,
         "skip_empty_fields": None,