Skip to content

Commit d8aaf67

Browse files
Fix tests.
1 parent a0e3170 commit d8aaf67

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

tcconfig/shaper/htb.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,9 +73,9 @@ def adjusted_burst_size(desired_burst: int, rate: int) -> int:
7373
else:
7474
adjusted_burst = desired_burst
7575
while adjusted_burst < (1 << 32):
76-
if calc_xmitsize_true(rate, calc_xmittime_bug(rate, burst)) >= desired_burst:
76+
if calc_xmitsize_true(rate, calc_xmittime_bug(rate, adjusted_burst)) >= desired_burst:
7777
return adjusted_burst
78-
burst += 1
78+
adjusted_burst += 1
7979
return adjusted_burst
8080

8181

test/test_burstsize.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import humanreadable as hr
66
import pytest
77

8-
from tcconfig.shaper.htb import adjusted_burst_size, desired_burst_size
8+
from tcconfig.shaper.htb import adjusted_burst_size, default_burst_size
99

1010

1111
@pytest.mark.parametrize(
@@ -18,10 +18,10 @@
1818
("80Gbps", 9000, 9010),
1919
],
2020
)
21-
def test_desired_burst(rate_hr, mtu, expected):
21+
def test_default_burst(rate_hr, mtu, expected):
2222
rate = hr.BitsPerSecond(rate_hr).byte_per_sec
23-
desired_burst = desired_burst_size(rate, mtu)
24-
assert desired_burst == expected
23+
default_burst = default_burst_size(rate, mtu)
24+
assert default_burst == expected
2525

2626

2727
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)