feat: add Size func for PriorityQueue

2025-10-04 07:06:30 +08:00 · 2022-06-05 17:29:36 +08:00
parent 61251fb0f1
commit a706d488e6
3 changed files with 628 additions and 0 deletions
--- a/datastructure/queue/priorityqueue_test.go
+++ b/datastructure/queue/priorityqueue_test.go
@@ -52,9 +52,13 @@ func TestPriorityQueue_Dequeue(t *testing.T) {
 		pq.Enqueue(i)
 	}

+	assert.Equal(10, pq.Size())
+
 	val, ok := pq.Dequeue()
 	assert.Equal(true, ok)
 	assert.Equal(10, val)

 	assert.Equal([]int{9, 8, 6, 7, 3, 2, 5, 1, 4}, pq.Data())
+
+	assert.Equal(9, pq.Size())
 }