feat(warm-up): quick checks

Author: TomMonks

content/13_warm_up.ipynb

@@ -7,7 +7,7 @@
    "source": [
     "# Implementing a warm-up period\n",
     "\n",
-    "We will implement warm-up as an event that resets all of our results collection variables.  \n",
+    "We will implement warm-up as a single event that resets all of our results collection variables.  \n",
     "\n",
     "This is a simpler approach than including lots of if statements in `simpy` processes."
    ]
@@ -309,7 +309,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 33,
    "id": "b3e686ce-5371-4471-a052-b9d43309bc85",
    "metadata": {},
    "outputs": [],
@@ -334,7 +334,7 @@
     "        inter_arrival_time = args.arrival_strokes.sample()\n",
     "        yield env.timeout(inter_arrival_time)\n",
     "\n",
-    "        args.results[\"n_arrivals\"] = patient_id\n",
+    "        args.results[\"n_arrivals\"] += 1\n",
     "        \n",
     "        trace(f\"{env.now:.2f}: Stroke arrival.\")\n",
     "\n",
@@ -352,7 +352,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 34,
    "id": "0d0ea6cf-7d95-4d2c-9690-fcdbdae35d84",
    "metadata": {},
    "outputs": [],
@@ -414,11 +414,84 @@
     "    return results"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "c13f5e57-723c-409b-a1ce-cdb831b4e166",
+   "metadata": {},
+   "source": [
+    "## Quick check 1: No warm-up"
+   ]
+  },
   {
    "cell_type": "code",
-   "execution_count": 23,
+   "execution_count": 35,
    "id": "caf52390-5455-4fa1-bb22-60b5b91ad8d0",
    "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "0.00: 🥵 Warm up complete.\n",
+      "3.29: Stroke arrival.\n",
+      "3.29: Patient 1 admitted to acute ward.(waited 0.00 days)\n",
+      "4.06: Stroke arrival.\n",
+      "4.06: Patient 2 admitted to acute ward.(waited 0.00 days)\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "{'mean_acute_wait': 0.0}"
+      ]
+     },
+     "execution_count": 35,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "TRACE = True\n",
+    "experiment = Experiment()\n",
+    "results = single_run(experiment, rep=0, wu_period=0.0, rc_period=5.0)\n",
+    "results"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 36,
+   "id": "ddedb4f1-207d-4295-9ae4-c49b2c7cdcaf",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'n_arrivals': 2, 'waiting_acute': [0.0, 0.0]}"
+      ]
+     },
+     "execution_count": 36,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# check how many patient waiting times recorded.\n",
+    "experiment.results"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "660ea2e1-d9c2-4355-876c-43dfd9dab0fe",
+   "metadata": {},
+   "source": [
+    "## Quick check 1: Include a warm-up"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 37,
+   "id": "72b5284a-1fcb-4126-b663-c0ef0002e4bf",
+   "metadata": {},
    "outputs": [
     {
      "name": "stdout",
@@ -443,7 +516,7 @@
        "{'mean_acute_wait': 0.0}"
       ]
      },
-     "execution_count": 23,
+     "execution_count": 37,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -457,17 +530,17 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 24,
-   "id": "ddedb4f1-207d-4295-9ae4-c49b2c7cdcaf",
+   "execution_count": 38,
+   "id": "7f5e282b-0f41-41df-bdca-f128e7d418c1",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "{'n_arrivals': 5, 'waiting_acute': [0.0, 0.0, 0.0]}"
+       "{'n_arrivals': 3, 'waiting_acute': [0.0, 0.0, 0.0]}"
       ]
      },
-     "execution_count": 24,
+     "execution_count": 38,
      "metadata": {},
      "output_type": "execute_result"
     }