Skip to content

Commit b118463

Browse files
committed
Added get_serialized_bytes
1 parent e49f685 commit b118463

2 files changed

Lines changed: 4 additions & 0 deletions

File tree

src/bloom_filter_wrapper.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,9 @@ void bind_bloom_filter(nb::module_ &m, const char* name) {
4848
.def("query", static_cast<bool (bloom_filter_type::*)(const std::string&) const>(&bloom_filter_type::query),
4949
nb::arg("item"),
5050
"Queries the filter for the given string")
51+
.def("get_serialized_size_bytes",
52+
[](const bloom_filter_type& sk) { return sk.get_serialized_size_bytes(); },
53+
"Returns the size in bytes of the serialized image of the filter")
5154
.def("serialize",
5255
[](const bloom_filter_type& sk) {
5356
auto v = sk.serialize(); // vector_bytes (std::vector<uint8_t, Allocator>)

tests/bloom_filter_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ def test_bloom_filter_serialize_deserialize(self):
6161

6262
payload = bf.serialize()
6363
self.assertTrue(len(payload) > 0)
64+
self.assertEqual(bf.get_serialized_size_bytes(), len(payload))
6465

6566
restored = bf.deserialize(payload)
6667
self.assertFalse(restored.is_empty())

0 commit comments

Comments
 (0)