[AT3P] Adjust number of quant units during encoding.

1 files changed, 16 insertions, 0 deletions

diff --git a/src/lib/bs_encode/encode_ut.cpp b/src/lib/bs_encode/encode_ut.cpp
index e9293a2..39f0ff1 100644
--- a/src/lib/bs_encode/encode_ut.cpp
+++ b/src/lib/bs_encode/encode_ut.cpp
@@ -47,6 +47,10 @@ public:
     void Dump(NBitStream::TBitStream& bs) override {
          EXPECT_EQ(EncCalls, ExpCalls);
     }
+
+    uint32_t GetConsumption() const noexcept override {
+        return 0;
+    }
 private:
     const size_t ExpCalls;
     size_t EncCalls = 0;
@@ -74,6 +78,10 @@ public:
              bs.Write(1, 1);
          }
     }
+
+    uint32_t GetConsumption() const noexcept override {
+        return 1 * Bits;
+    }
 private:
     const size_t ExpCalls;
     size_t EncCalls = 0;
@@ -94,6 +102,10 @@ public:
     void Dump(NBitStream::TBitStream& bs) override {
          EXPECT_EQ(EncCalls, ExpCalls);
     }
+
+    uint32_t GetConsumption() const noexcept override {
+        return 0;
+    }
 private:
     const size_t ExpCalls;
     size_t EncCalls = 0;
@@ -115,6 +127,10 @@ public:
     void Dump(NBitStream::TBitStream& bs) override {
          EXPECT_EQ(EncCalls, 1);
     }
+
+    uint32_t GetConsumption() const noexcept override {
+        return 0;
+    }
 private:
     size_t EncCalls = 0;
 };