From fe14c24a8953a991816d98fa072e5ce43bd1b93e Mon Sep 17 00:00:00 2001
From: Stuckinaboot <aspyn@seas.upenn.edu>
Date: Sun, 7 Apr 2019 14:46:01 -0400
Subject: [PATCH] Add working multi-topic test

---
 examples/sharding/driver.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/examples/sharding/driver.py b/examples/sharding/driver.py
index 4989d49..7dd4669 100644
--- a/examples/sharding/driver.py
+++ b/examples/sharding/driver.py
@@ -72,7 +72,9 @@ async def main():
     
     # Define topic map
     topic_map = {
-        "1": [0, 1, 2, 3, 4, 5]
+        "1": [0, 1],
+        "2": [2, 3],
+        "3": [4, 5]
     }
 
     topics = topic_map.keys()