14
14
* See the License for the specific language governing permissions and
15
15
* limitations under the License.
16
16
*/
17
- #include < tests/utils/gluten_test_util.h>
18
- #include < iostream>
17
+ #include < zlib.h>
19
18
#include < Core/Settings.h>
20
19
#include < Interpreters/Context.h>
21
20
#include < Parser/SerializedPlanParser.h>
22
- #include < gtest/gtest.h>
23
- #include < Common/QueryContext.h>
24
- #include < zlib.h>
21
+ #include < Storages/SubstraitSource/Delta/Bitmap/DeltaDVRoaringBitmapArray.h>
25
22
#include < Storages/SubstraitSource/ReadBufferBuilder.h>
23
+ #include < gtest/gtest.h>
24
+ #include < tests/utils/gluten_test_util.h>
26
25
#include < roaring.hh>
27
- #include < Storages/SubstraitSource/Delta/Bitmap/DeltaDVRoaringBitmapArray .h>
26
+ #include < Common/QueryContext .h>
28
27
29
28
namespace DB ::Setting
30
29
{
@@ -136,11 +135,11 @@ TEST(Delta_DV, DeltaDVRoaringBitmapArray)
136
135
{
137
136
const auto context = DB::Context::createCopy (QueryContext::globalContext ());
138
137
139
- const Poco::URI file_uri (test::gtest_data (" deletion_vector_multiple.bin" ));
140
- const Poco::URI file_uri1 (test::gtest_data (" deletion_vector_only_one.bin" ));
138
+ const std::string file_uri (test::gtest_uri (" deletion_vector_multiple.bin" ));
139
+ const std::string file_uri1 (test::gtest_uri (" deletion_vector_only_one.bin" ));
141
140
142
- local_engine:: DeltaDVRoaringBitmapArray bitmap_array (context);
143
- bitmap_array.rb_read (file_uri. getPath () , 426433 , 426424 );
141
+ DeltaDVRoaringBitmapArray bitmap_array (context);
142
+ bitmap_array.rb_read (file_uri, 426433 , 426424 );
144
143
EXPECT_TRUE (bitmap_array.rb_contains (5 ));
145
144
EXPECT_TRUE (bitmap_array.rb_contains (3618 ));
146
145
EXPECT_TRUE (bitmap_array.rb_contains (155688 ));
@@ -158,8 +157,8 @@ TEST(Delta_DV, DeltaDVRoaringBitmapArray)
158
157
bitmap_array.rb_add (10000000000 );
159
158
EXPECT_TRUE (bitmap_array.rb_contains (10000000000 ));
160
159
161
- local_engine:: DeltaDVRoaringBitmapArray bitmap_array1 (context);
162
- bitmap_array1.rb_read (file_uri1. getPath () , 1 , 539 );
160
+ DeltaDVRoaringBitmapArray bitmap_array1 (context);
161
+ bitmap_array1.rb_read (file_uri1, 1 , 539 );
163
162
EXPECT_TRUE (bitmap_array1.rb_contains (0 ));
164
163
EXPECT_TRUE (bitmap_array1.rb_contains (1003 ));
165
164
EXPECT_TRUE (bitmap_array1.rb_contains (880 ));
@@ -172,10 +171,10 @@ TEST(Delta_DV, DeltaDVRoaringBitmapArray)
172
171
EXPECT_TRUE (bitmap_array.rb_contains (1003 ));
173
172
EXPECT_TRUE (bitmap_array.rb_contains (880 ));
174
173
175
- const Poco::URI file_uri2 (test::gtest_data (" deletion_vector_long_values.bin" ));
174
+ const std::string file_uri2 (test::gtest_uri (" deletion_vector_long_values.bin" ));
176
175
177
- local_engine:: DeltaDVRoaringBitmapArray bitmap_array2 (context);
178
- bitmap_array2.rb_read (file_uri2. getPath () , 1 , 4047 );
176
+ DeltaDVRoaringBitmapArray bitmap_array2 (context);
177
+ bitmap_array2.rb_read (file_uri2, 1 , 4047 );
179
178
EXPECT_FALSE (bitmap_array2.rb_is_empty ());
180
179
EXPECT_EQ (2098 , bitmap_array2.rb_size ());
181
180
EXPECT_TRUE (bitmap_array2.rb_contains (0 ));
0 commit comments