diff --git a/packages/cli-kit/src/public/common/collection.test.ts b/packages/cli-kit/src/public/common/collection.test.ts
index 605c6465582..f33685f7563 100644
--- a/packages/cli-kit/src/public/common/collection.test.ts
+++ b/packages/cli-kit/src/public/common/collection.test.ts
@@ -20,6 +20,19 @@ describe('groupBy', () => {
       Barcelona: [{city: 'Barcelona', name: 'User3'}],
     })
   })
+
+  test('groups elements using a function iteratee', () => {
+    expect(groupBy([6.1, 4.2, 6.3], Math.floor)).toEqual({
+      '4': [4.2],
+      '6': [6.1, 6.3],
+    })
+  })
+
+  test('handles empty, null, or undefined collection', () => {
+    expect(groupBy([], 'city')).toEqual({})
+    expect(groupBy(null, 'city')).toEqual({})
+    expect(groupBy(undefined, 'city')).toEqual({})
+  })
 })
 
 describe('partition', () => {
@@ -43,4 +56,10 @@ describe('partition', () => {
       ],
     ])
   })
+
+  test('handles empty, null, or undefined collection', () => {
+    expect(partition([], () => true)).toEqual([[], []])
+    expect(partition(null, () => true)).toEqual([[], []])
+    expect(partition(undefined, () => true)).toEqual([[], []])
+  })
 })