Modify test according to comments

microsoft · Jun 21, 2022 · 08744df · 08744df
1 parent c4068e3
commit 08744df
Show file tree

Hide file tree

Showing 6 changed files with 672 additions and 24 deletions.
diff --git a/maro/simulator/scenarios/supply_chain/units/distribution.py b/maro/simulator/scenarios/supply_chain/units/distribution.py
@@ -100,6 +100,7 @@ def initialize(self) -> None:
             self._busy_vehicle_num[vehicle_type] = 0
 
             # TODO: add vehicle patient setting if needed
+
         self.data_model.initialize()
 
         for sku_id in self.facility.products.keys():

diff --git a/tests/data/supply_chain/case_01/config.yml b/tests/data/supply_chain/case_01/config.yml
@@ -88,6 +88,7 @@ world:
         sku3:
           init_stock: 80
           has_manufacture: True
+          has_consumer: True
           max_manufacture_rate: 50
           manufacture_leading_time: 1
           unit_product_cost: 1
@@ -105,6 +106,7 @@ world:
           init_stock: 96
           has_manufacture: True
           max_manufacture_rate: 50
+          has_consumer: True
           manufacture_leading_time: 1
           unit_product_cost: 1
           price: 100

diff --git a/tests/data/supply_chain/case_05/config.yml b/tests/data/supply_chain/case_05/config.yml
@@ -146,6 +146,25 @@ world:
         unit_delay_order_penalty: 20
         unit_order_cost: 0
 
+    - name: "Supplier_SKU1"
+      definition_ref: "SupplierFacility"
+      skus:
+        sku3:
+          init_stock: 80
+          has_manufacture: True
+          max_manufacture_rate: 50
+          manufacture_leading_time: 1
+          unit_product_cost: 1
+          price: 10
+          unit_delay_order_penalty: 10
+          has_consumer: True
+      children:
+        storage: *small_storage
+        distribution: *normal_distribution
+      config:
+        unit_delay_order_penalty: 20
+        unit_order_cost: 0
+
     - name: "Warehouse_001"
       definition_ref: "WarehouseFacility"
       skus:
@@ -232,7 +251,7 @@ world:
         storage: *single_storage
       config:
         unit_order_cost: 200
-        file_path: "tests/data/supply_chain/case_05/test_case_05.csv"
+        file_path: "tests/data/supply_chain/case_04/test_case_04.csv"
 
   topology:
 
@@ -258,6 +277,10 @@ world:
 
     Warehouse_001:
       sku3:
+        "Supplier_SKU1":
+          "train":
+            vlt: 3
+            cost: 1
         "Supplier_SKU3":
           "train":
             vlt: 3