pandas-dev#23049: test for Fatal Stack Overflow stemming From Misuse …

…of astype('category') (pandas-dev#25366)
Pingviinituutti · Feb 28, 2019 · c56c1be · c56c1be
1 parent 7578a0c
commit c56c1be
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pandas/tests/frame/test_combine_concat.py b/pandas/tests/frame/test_combine_concat.py
@@ -504,6 +504,16 @@ def test_concat_numerical_names(self):
                                                                 names=[1, 2]))
         tm.assert_frame_equal(result, expected)
 
+    def test_concat_astype_dup_col(self):
+        # gh 23049
+        df = pd.DataFrame([{'a': 'b'}])
+        df = pd.concat([df, df], axis=1)
+
+        result = df.astype('category')
+        expected = pd.DataFrame(np.array(["b", "b"]).reshape(1, 2),
+                                columns=["a", "a"]).astype("category")
+        tm.assert_frame_equal(result, expected)
+
 
 class TestDataFrameCombineFirst(TestData):