fix summary return

OurPlanscape · Aug 30, 2024 · 2926648 · 2926648
1 parent cfca24c
commit 2926648
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/planscape/impacts/services.py b/src/planscape/impacts/services.py
@@ -185,6 +185,7 @@ def generate_summary(
                     "type": p["type"],
                     "treated_stand_count": p["treated_stand_count"],
                     "area_acres": p["treated_stand_count"] * stand_area,
+                    "stand_ids": p["stand_ids"],
                 }
                 for p in filter(
                     lambda p: p["project_area__id"] == project.id,

diff --git a/src/planscape/impacts/tests/test_services.py b/src/planscape/impacts/tests/test_services.py
@@ -179,6 +179,8 @@ def test_summary_is_returned_correctly(self):
         )[0]
         self.assertIn("prescriptions", proj_area_1)
         self.assertEqual(len(proj_area_1["prescriptions"]), 1)
+        stands1 = proj_area_1["prescriptions"][0]["stand_ids"]
+        self.assertGreater(len(stands1), 0)
         self.assertIn("prescriptions", proj_area_2)
         self.assertEqual(len(proj_area_2["prescriptions"]), 1)