Added reset teset in supply chain scenarios

maro/simulator/scenarios/supply_chain/datamodels/distribution.py

@@ -19,5 +19,8 @@ class DistributionDataModel(DataModelBase):
     def __init__(self) -> None:
         super(DistributionDataModel, self).__init__()
 
+    def initialize(self) -> None:
+        self.reset()
+
     def reset(self) -> None:
         super(DistributionDataModel, self).reset()

maro/simulator/scenarios/supply_chain/units/distribution.py

@@ -101,6 +101,8 @@ def initialize(self) -> None:
 
             # TODO: add vehicle patient setting if needed
 
+        self.data_model.initialize()
+
         for sku_id in self.facility.products.keys():
             self._unit_delay_order_penalty[sku_id] = self.facility.skus[sku_id].unit_delay_order_penalty
 

tests/data/supply_chain/case_04/test_case_04.csv

@@ -4,11 +4,23 @@ food_1,2021/1/2,43.1,33.39,20
 food_1,2021/1/3,43.2,33.39,30
 food_1,2021/1/4,43.3,33.39,40
 food_1,2021/1/5,43.4,33.39,50
+food_1,2021/1/6,43.4,33.39,60
+food_1,2021/1/7,43.4,33.39,70
+food_1,2021/1/8,43.4,33.39,80
+food_1,2021/1/9,43.4,33.39,90
+food_1,2021/1/10,43.4,33.39,100
+food_1,2021/1/11,43.4,33.39,110
 hobby_1,2021/1/1,28.32,21.79,100
 hobby_1,2021/1/2,28.32,21.79,200
 hobby_1,2021/1/3,28.32,21.79,300
 hobby_1,2021/1/4,28.32,21.79,400
 hobby_1,2021/1/5,28.32,21.79,500
+hobby_1,2021/1/6,28.32,21.79,600
+hobby_1,2021/1/7,28.32,21.79,700
+hobby_1,2021/1/8,28.32,21.79,800
+hobby_1,2021/1/9,28.32,21.79,900
+hobby_1,2021/1/10,28.32,21.79,1000
+hobby_1,2021/1/11,28.32,21.79,1100
 household_1,2022/5/14,17.35974446,14.4,146
 household_1,2022/5/15,17.29592816,14.4,164
 household_1,2022/5/16,17.43129521,14.4,127

tests/data/supply_chain/case_05/config.yml

@@ -146,6 +146,30 @@ world:
         unit_delay_order_penalty: 20
         unit_order_cost: 0
 
+    - name: "Supplier_SKU1"
+      definition_ref: "SupplierFacility"
+      skus:
+        sku1:
+          init_stock: 20
+          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
+        sku3:
+          init_stock: 80
+          unit_product_cost: 1
+          price: 10
+          unit_delay_order_penalty: 10
+      children:
+        storage: *small_storage
+        distribution: *normal_distribution
+      config:
+        unit_delay_order_penalty: 20
+        unit_order_cost: 0
+
     - name: "Warehouse_001"
       definition_ref: "WarehouseFacility"
       skus:
@@ -154,6 +178,7 @@ world:
           sub_storage_id: 1
           storage_upper_bound: 40
           price: 100
+          has_consumer: True
         sku2:
           init_stock: 12
           sub_storage_id: 1
@@ -257,6 +282,11 @@ world:
             cost: 0.6
 
     Warehouse_001:
+      sku1:
+        "Supplier_SKU1":
+          "train":
+            vlt: 3
+            cost: 1
       sku3:
         "Supplier_SKU3":
           "train":