From 129e4c42dafd40ec9c867bb9b62716eee3c47a32 Mon Sep 17 00:00:00 2001 From: alanprot Date: Wed, 11 Dec 2024 17:12:19 -0800 Subject: [PATCH] Change query fuzz test for bottomk, topk: take 2 Signed-off-by: alanprot --- integration/query_fuzz_test.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/integration/query_fuzz_test.go b/integration/query_fuzz_test.go index 72ae8ce3309..1e1485f2197 100644 --- a/integration/query_fuzz_test.go +++ b/integration/query_fuzz_test.go @@ -312,6 +312,11 @@ func TestDisableChunkTrimmingFuzz(t *testing.T) { t.Logf("case %d error mismatch.\n%s: %s\nerr1: %v\nerr2: %v\n", i, qt, tc.query, tc.err1, tc.err2) failures++ } + } else if shouldUseSampleNumComparer(tc.query) { + if !cmp.Equal(tc.res1, tc.res2, sampleNumComparer) { + t.Logf("case %d # of samples mismatch.\n%s: %s\nres1: %s\nres2: %s\n", i, qt, tc.query, tc.res1.String(), tc.res2.String()) + failures++ + } } else if !cmp.Equal(tc.res1, tc.res2, comparer) { t.Logf("case %d results mismatch.\n%s: %s\nres1: %s\nres2: %s\n", i, qt, tc.query, tc.res1.String(), tc.res2.String()) failures++ @@ -508,6 +513,11 @@ func TestExpandedPostingsCacheFuzz(t *testing.T) { t.Logf("case %d error mismatch.\n%s: %s\nerr1: %v\nerr2: %v\n", i, qt, tc.query, tc.err1, tc.err2) failures++ } + } else if shouldUseSampleNumComparer(tc.query) { + if !cmp.Equal(tc.res1, tc.res2, sampleNumComparer) { + t.Logf("case %d # of samples mismatch.\n%s: %s\nres1: %s\nres2: %s\n", i, qt, tc.query, tc.res1.String(), tc.res2.String()) + failures++ + } } else if !cmp.Equal(tc.res1, tc.res2, comparer) { t.Logf("case %d results mismatch.\n%s: %s\nres1: %s\nres2: %s\n", i, qt, tc.query, tc.res1.String(), tc.res2.String()) failures++