From 087c78a0dfb98e2598cb48c54331a53ba31576b0 Mon Sep 17 00:00:00 2001
From: Travis <travis@pilosa.com>
Date: Thu, 22 Jun 2017 12:04:01 -0500
Subject: [PATCH] add test coverage for `CountRange` operations

---
 roaring/roaring_test.go | 29 ++++++++++++++++++++++++++++-
 1 file changed, 28 insertions(+), 1 deletion(-)

diff --git a/roaring/roaring_test.go b/roaring/roaring_test.go
index a097e1fbe..c9b18c9a2 100644
--- a/roaring/roaring_test.go
+++ b/roaring/roaring_test.go
@@ -163,7 +163,8 @@ func TestBitmap_Max(t *testing.T) {
 		}
 	}
 }
-func TestBitmap_CountRange(t *testing.T) {
+
+func TestBitmap_BitmapCountRange(t *testing.T) {
 	bm0 := roaring.NewBitmap(0, 2683177)
 	for i := uint64(628); i < 2683301; i++ {
 		bm0.Add(i)
@@ -184,6 +185,32 @@ func TestBitmap_CountRange(t *testing.T) {
 	if n := bm0.CountRange(0, 1); n != 1 {
 		t.Fatalf("unexpected n: %d", n)
 	}
+
+	// Test the case where the range is outside of the bitmap space.
+	if n := bm0.CountRange(10000000, 10000001); n != 0 {
+		t.Fatalf("unexpected n: %d", n)
+	}
+}
+
+func TestBitmap_ArrayCountRange(t *testing.T) {
+	bm0 := roaring.NewBitmap(0, 2683177, 2683313)
+	if n := bm0.CountRange(1, 2683311); n != 1 {
+		t.Fatalf("unexpected n: %d", n)
+	}
+}
+
+func TestBitmap_RunCountRange(t *testing.T) {
+	bm0 := roaring.NewBitmap(0, 1, 2, 3, 4, 5, 12, 13, 14, 15, 16, 17, 1000000, 1000002, 1000003, 1000004, 1000005, 1000006, 1000010, 1000011, 1000012, 1000013, 1000014)
+	bm0.Optimize() // convert to runs
+	if n := bm0.CountRange(15, 1000003); n != 5 {
+		t.Fatalf("unexpected n: %d", n)
+	}
+
+	bm1 := roaring.NewBitmap(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17)
+	bm1.Optimize() // convert to runs
+	if n := bm1.CountRange(5, 12); n != 7 {
+		t.Fatalf("unexpected n: %d", n)
+	}
 }
 
 func TestBitmap_Intersection(t *testing.T) {