Skip to content

Commit

Permalink
lint fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
udovichenkoAlexander committed Nov 5, 2024
1 parent f9e4ff5 commit 9b90ddb
Showing 1 changed file with 20 additions and 20 deletions.
40 changes: 20 additions & 20 deletions tests/util/test_wheel_timer.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,55 +28,55 @@ class WheelTimerTestCase(unittest.TestCase):
def test_single_insert_fetch(self) -> None:
wheel: WheelTimer[object] = WheelTimer(bucket_size=5)

wheel.insert(100, '1', 150)
wheel.insert(100, "1", 150)

self.assertListEqual(wheel.fetch(101), [])
self.assertListEqual(wheel.fetch(110), [])
self.assertListEqual(wheel.fetch(120), [])
self.assertListEqual(wheel.fetch(130), [])
self.assertListEqual(wheel.fetch(149), [])
self.assertListEqual(wheel.fetch(156), ['1'])
self.assertListEqual(wheel.fetch(156), ["1"])
self.assertListEqual(wheel.fetch(170), [])

def test_multi_insert(self) -> None:
wheel: WheelTimer[object] = WheelTimer(bucket_size=5)

wheel.insert(100, '1', 150)
wheel.insert(105, '2', 130)
wheel.insert(106, '3', 160)
wheel.insert(100, "1", 150)
wheel.insert(105, "2", 130)
wheel.insert(106, "3", 160)

self.assertListEqual(wheel.fetch(110), [])
self.assertListEqual(wheel.fetch(135), ['2'])
self.assertListEqual(wheel.fetch(135), ["2"])
self.assertListEqual(wheel.fetch(149), [])
self.assertListEqual(wheel.fetch(158), ['1'])
self.assertListEqual(wheel.fetch(158), ["1"])
self.assertListEqual(wheel.fetch(160), [])
self.assertListEqual(wheel.fetch(200), ['3'])
self.assertListEqual(wheel.fetch(200), ["3"])
self.assertListEqual(wheel.fetch(210), [])

def test_insert_past(self) -> None:
wheel: WheelTimer[object] = WheelTimer(bucket_size=5)

wheel.insert(100, '1', 50)
self.assertListEqual(wheel.fetch(120), ['1'])
wheel.insert(100, "1", 50)
self.assertListEqual(wheel.fetch(120), ["1"])

def test_insert_past_multi(self) -> None:
wheel: WheelTimer[object] = WheelTimer(bucket_size=5)

wheel.insert(100, '1', 150)
wheel.insert(100, '2', 140)
wheel.insert(100, '3', 50)
self.assertListEqual(wheel.fetch(110), ['3'])
wheel.insert(100, "1", 150)
wheel.insert(100, "2", 140)
wheel.insert(100, "3", 50)
self.assertListEqual(wheel.fetch(110), ["3"])
self.assertListEqual(wheel.fetch(120), [])
self.assertListEqual(wheel.fetch(147), ['2'])
self.assertListEqual(wheel.fetch(200), ['1'])
self.assertListEqual(wheel.fetch(147), ["2"])
self.assertListEqual(wheel.fetch(200), ["1"])
self.assertListEqual(wheel.fetch(240), [])

def test_multi_insert_then_past(self) -> None:
wheel: WheelTimer[object] = WheelTimer(bucket_size=5)

wheel.insert(100, '1', 150)
wheel.insert(100, '2', 160)
wheel.insert(100, '3', 155)
wheel.insert(100, "1", 150)
wheel.insert(100, "2", 160)
wheel.insert(100, "3", 155)

self.assertListEqual(wheel.fetch(110), [])
self.assertListEqual(wheel.fetch(158), ['1'])
self.assertListEqual(wheel.fetch(158), ["1"])

0 comments on commit 9b90ddb

Please sign in to comment.