fix

Author: pauldambra

posthog/test/test_size_limited_dict.py

@@ -0,0 +1,37 @@
+import unittest
+from dataclasses import dataclass
+from datetime import date, datetime, timedelta
+from decimal import Decimal
+from typing import Optional
+from uuid import UUID
+
+import six
+from dateutil.tz import tzutc
+from parameterized import parameterized
+from pydantic import BaseModel
+from pydantic.v1 import BaseModel as BaseModelV1
+
+from posthog import utils
+
+
+class TestSizeLimitedDict(unittest.TestCase):
+    @parameterized.expand([
+        (10, 100),
+        (5, 20),
+        (20, 200)
+    ])
+    def test_size_limited_dict(self, size: int, iterations: int) -> None:
+        values = utils.SizeLimitedDict(size, lambda _: -1)
+
+        for i in range(iterations):
+            values[i] = i
+
+            assert values[i] == i
+            assert len(values) == i % size + 1
+
+            if i % size == 0:
+                # old numbers should've been removed
+                self.assertIsNone(values.get(i - 1))
+                self.assertIsNone(values.get(i - 3))
+                self.assertIsNone(values.get(i - 5))
+                self.assertIsNone(values.get(i - 9))

posthog/test/test_utils.py

@@ -170,37 +170,3 @@ class TestDataClass:
                 "inner_optional": None,
             },
         }
-
-
-class TestSizeLimitedDict(unittest.TestCase):
-    @parameterized.expand(
-        [
-            (5, 20),
-            (10, 100),
-            (3, 15),
-        ]
-    )
-    def test_size_limited_dict_with_different_sizes(self, size, iterations):
-        values = utils.SizeLimitedDict(size, lambda _: -1)
-
-        for i in range(iterations):
-            values[i] = i
-            assert values[i] == i
-            assert len(values) == min(i + 1, size)
-
-    def test_size_limited_dict(self):
-        size = 10
-        values = utils.SizeLimitedDict(size, lambda _: -1)
-
-        for i in range(100):
-            values[i] = i
-
-            assert values[i] == i
-            assert len(values) == i % size + 1
-
-            if i % size == 0:
-                # old numbers should've been removed
-                assert values.get(i - 1) is None
-                assert values.get(i - 3) is None
-                assert values.get(i - 5) is None
-                assert values.get(i - 9) is None