From 31f1fe0be13568eced6ae550307d16d233fb877a Mon Sep 17 00:00:00 2001 From: Travis Date: Fri, 25 Aug 2017 19:23:09 -0500 Subject: [PATCH] set container.n in differenceRunRun --- roaring/roaring.go | 1 + roaring/roaring_internal_test.go | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/roaring/roaring.go b/roaring/roaring.go index fa744e829..77dfd8803 100644 --- a/roaring/roaring.go +++ b/roaring/roaring.go @@ -2621,6 +2621,7 @@ func differenceRunRun(a, b *container) *container { } } + output.n = output.count() return output } diff --git a/roaring/roaring_internal_test.go b/roaring/roaring_internal_test.go index c0caed990..5d1a985d1 100644 --- a/roaring/roaring_internal_test.go +++ b/roaring/roaring_internal_test.go @@ -1601,6 +1601,7 @@ func TestDifferenceRunRun(t *testing.T) { aruns []interval16 bruns []interval16 exp []interval16 + expn int }{ { // this tests all six overlap combinations @@ -1609,6 +1610,7 @@ func TestDifferenceRunRun(t *testing.T) { aruns: []interval16{{start: 3, last: 6}, {start: 13, last: 16}, {start: 24, last: 26}, {start: 33, last: 38}, {start: 43, last: 46}, {start: 53, last: 56}}, bruns: []interval16{{start: 1, last: 8}, {start: 11, last: 14}, {start: 21, last: 23}, {start: 35, last: 37}, {start: 44, last: 48}, {start: 57, last: 59}}, exp: []interval16{{start: 15, last: 16}, {start: 24, last: 26}, {start: 33, last: 34}, {start: 38, last: 38}, {start: 43, last: 43}, {start: 53, last: 56}}, + expn: 13, }, } for i, test := range tests { @@ -1620,6 +1622,9 @@ func TestDifferenceRunRun(t *testing.T) { if !reflect.DeepEqual(ret.runs, test.exp) { t.Fatalf("test #%v expected %v, but got %v", i, test.exp, ret.runs) } + if ret.n != test.expn { + t.Fatalf("test #%v expected n=%v, but got n=%v", i, test.expn, ret.n) + } } }