fix: preserve scalar items in list flatten

Fixes #5705
2026-09-04 23:17:20 +08:00 · 2026-05-14 22:18:59 +08:00
parent f50773711e
commit 9128955bf9
2 changed files with 20 additions and 1 deletions
@@ -0,0 +1,15 @@
+from unittest import TestCase
+
+from app.utils.structures import ListUtils
+
+
+class ListUtilsTest(TestCase):
+    def test_flatten_keeps_scalar_items_in_mixed_list(self):
+        self.assertEqual(ListUtils.flatten([1, [2, 3], 4]), [1, 2, 3, 4])
+
+    def test_flatten_returns_plain_list_unchanged(self):
+        source = [1, 2, 3]
+        self.assertEqual(ListUtils.flatten(source), source)
+
+    def test_flatten_rejects_non_list_input(self):
+        self.assertEqual(ListUtils.flatten("1,2,3"), [])