diff --git a/artifact_1_bandwidth_analysis/tests/data/sample_affinity_enforced.json b/artifact_1_bandwidth_analysis/tests/data/sample_affinity_enforced.json
new file mode 100644
index 0000000..2a1f62c
--- /dev/null
+++ b/artifact_1_bandwidth_analysis/tests/data/sample_affinity_enforced.json
@@ -0,0 +1,22 @@
+[
+  {
+    "worker_id": "A1",
+    "timestamp": 0.5
+  },
+  {
+    "worker_id": "A1",
+    "timestamp": 1.2
+  },
+  {
+    "worker_id": "A2",
+    "timestamp": 0.9
+  },
+  {
+    "worker_id": "A2",
+    "timestamp": 1.4
+  },
+  {
+    "worker_id": "A3",
+    "timestamp": 0.7
+  }
+]
\ No newline at end of file