update code sample for pandas-dev#38523

Author: simonjayhawkins

bisect/38523.py

@@ -1,3 +1,5 @@
+import sys
+
 import pandas as pd
 
 print(pd.__version__)
@@ -12,16 +14,25 @@
 result = df.groupby(level=0)["Max Speed"].rolling(2).sum()
 print(result)
 
+# expected = pd.MultiIndex.from_tuples(
+#     [
+#         ("Falcon", "Falcon", "Captive"),
+#         ("Falcon", "Falcon", "Wild"),
+#         ("Parrot", "Parrot", "Captive"),
+#         ("Parrot", "Parrot", "Wild"),
+#     ],
+#     names=["Animal", "Animal", "Type"],
+# )
+
 expected = pd.MultiIndex.from_tuples(
-    [
-        ("Falcon", "Falcon", "Captive"),
-        ("Falcon", "Falcon", "Wild"),
-        ("Parrot", "Parrot", "Captive"),
-        ("Parrot", "Parrot", "Wild"),
-    ],
-    names=["Animal", "Animal", "Type"],
+    [("Falcon",), ("Falcon",), ("Parrot",), ("Parrot",)], names=["Animal"]
 )
 
 import pandas.testing as tm
 
-tm.assert_index_equal(result.index, expected)
+try:
+    tm.assert_index_equal(result.index, expected)
+except AssertionError:
+    pass
+else:
+    sys.exit(1)