From f85fd63c74d1b2a500ed4ad9a546a6df5ff2ed9b Mon Sep 17 00:00:00 2001
From: Gregory <neogen570@gmail.com>
Date: Sun, 2 Jan 2022 23:55:38 +0200
Subject: [PATCH] multiway merge

---
 .../index_priority_queue_test.go              | 33 +++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/sorting/priority_queue/index_priority_queue_test.go b/sorting/priority_queue/index_priority_queue_test.go
index f98faaf..d0759ae 100644
--- a/sorting/priority_queue/index_priority_queue_test.go
+++ b/sorting/priority_queue/index_priority_queue_test.go
@@ -1,6 +1,8 @@
 package priority_queue
 
 import (
+	"io"
+	"strings"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -156,3 +158,34 @@ func TestIndexChange(t *testing.T) {
 
 	assert.Equal(t, 10, pq.top())
 }
+
+func TestMultiwayMerge(t *testing.T) {
+	// ordered "streams"
+	firstStream := strings.NewReader("ABCFGIIZ")
+	secondStream := strings.NewReader("BDHPQQ")
+	thirdStream := strings.NewReader("ABEFJN")
+
+	allStreams := []*strings.Reader{firstStream, secondStream, thirdStream}
+
+	expected := "AABBBCDEFFGHIIJNPQQZ"
+	actual := ""
+
+	pq := NewIPQ(func(t1, t2 string) bool { return t1 > t2 }, len(allStreams)+1)
+
+	for i, stream := range allStreams {
+		rune, _, _ := stream.ReadRune()
+		pq.insert(i+1, string(rune))
+	}
+
+	for !pq.isEmpty() {
+		actual += string(pq.top())
+		streamIndex := pq.topKey()
+		pq.remove()
+		rune, _, err := allStreams[streamIndex-1].ReadRune()
+		if err != io.EOF {
+			pq.insert(streamIndex, string(rune))
+		}
+	}
+
+	assert.Equal(t, expected, actual)
+}