seedgraph 0.2.0__tar.gz → 0.2.1__tar.gz

seedgraph-0.2.1/.hypothesis/constants/087165afdcb0189b

@@ -0,0 +1,4 @@
+# file: /Users/rachidj/DEV/seedgraph/src/seedgraph/sampling.py
+# hypothesis_version: 6.170.9
+
+['__use_weighting__', 'a', 'b', 'c', 'elements', 'factories', 'generator', 'length', 'providers', 'random', 'self', 'unique', 'use_weighting']

seedgraph-0.2.1/.hypothesis/constants/81538bbb42f71e5a

@@ -0,0 +1,4 @@
+# file: /Users/rachidj/DEV/seedgraph/src/seedgraph/__init__.py
+# hypothesis_version: 6.170.9
+
+['0.2.0', 'AmbiguousParentError', 'Graph', 'IncoherentGraphError', 'SeedgraphError', 'UnknownShapeKeyError', '__version__', 'seed', 'seed_async', 'sqlite', 'unreachable']

seedgraph-0.2.1/.hypothesis/constants/e5efc23d4d56e78c

@@ -0,0 +1,4 @@
+# file: /Users/rachidj/DEV/seedgraph/src/seedgraph/sampling.py
+# hypothesis_version: 6.170.9
+
+['__use_weighting__', 'a', 'b', 'c', 'elements', 'factories', 'generator', 'length', 'providers', 'random', 'self', 'unique', 'use_weighting']

{seedgraph-0.2.0 → seedgraph-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: seedgraph
-Version: 0.2.0
+Version: 0.2.1
 Summary: Seed your SQLAlchemy models as a referentially-consistent graph — one call, shared parents, verified links.
 Project-URL: Homepage, https://github.com/jrachid/seedgraph
 Project-URL: Documentation, https://jrachid.github.io/seedgraph/

{seedgraph-0.2.0 → seedgraph-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "seedgraph"
-version = "0.2.0"
+version = "0.2.1"
 description = "Seed your SQLAlchemy models as a referentially-consistent graph — one call, shared parents, verified links."
 readme = "README.md"
 license = "MIT"

{seedgraph-0.2.0 → seedgraph-0.2.1}/src/seedgraph/__init__.py

@@ -45,7 +45,7 @@ from seedgraph.verification import IncoherentGraphError, verify_graph
 if TYPE_CHECKING:
     from sqlalchemy.ext.asyncio import AsyncSession
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 
 __all__ = [
     "AmbiguousParentError",

{seedgraph-0.2.0 → seedgraph-0.2.1}/src/seedgraph/sampling.py

@@ -1,4 +1,4 @@
-"""Draw from Faker's weighted tables through cumulative weights computed once, with the exact values Faker itself draws."""
+"""Draw from the weighted tables Faker ships through cumulative weights computed once, with the exact values Faker itself draws."""
 
 import inspect
 from collections import OrderedDict
@@ -12,12 +12,15 @@ from faker.providers import BaseProvider
 
 __all__ = ["install_cumulative_sampling"]
 
-CACHE_ATTRIBUTE = "_seedgraph_cumulative"
 FAKER_SIGNATURE = ["self", "elements", "length", "unique", "use_weighting"]
 
+# The tables the providers carry as class attributes, by id: only these never change, so only these keep cached weights.
+_shipped: dict[int, OrderedDict[Any, float]] = {}
+_cumulative: dict[int, tuple[tuple[Any, ...], list[float]]] = {}
+
 
 def install_cumulative_sampling(fake: Faker) -> None:
-    """Give each provider of this Faker a random_elements that reuses each weighted table's cumulative weights.
+    """Give each provider of this Faker a random_elements that reuses the cumulative weights of the tables Faker ships.
 
     Installs nothing when one of the Faker internals it relies on is missing.
     """
@@ -27,6 +30,8 @@ def install_cumulative_sampling(fake: Faker) -> None:
     if not all(hasattr(p, "__use_weighting__") and hasattr(getattr(p, "generator", None), "random") for p in providers):
         return
     for provider in providers:
+        for cls in type(provider).__mro__:
+            _shipped.update((id(table), table) for table in vars(cls).values() if isinstance(table, OrderedDict))
         provider.random_elements = MethodType(_random_elements, provider)
 
 
@@ -38,20 +43,20 @@ def _random_elements(
     use_weighting: bool | None = None,
 ) -> Sequence[Any]:
     weighting = provider.__use_weighting__ if use_weighting is None else use_weighting
-    if unique or not weighting or not isinstance(elements, OrderedDict):
+    shipped = _shipped.get(id(elements))
+    if unique or not weighting or shipped is None or shipped is not elements:
         return BaseProvider.random_elements(provider, elements, length, unique, use_weighting)
     rng = provider.generator.random
     if length is None:
-        length = rng.randint(1, len(elements))
-    keys, cumulative = _cumulative_table(elements)
+        length = rng.randint(1, len(shipped))
+    keys, cumulative = _cumulative_table(shipped)
     # Given cum_weights, random.choices skips its accumulate pass and bisects the very totals it would have built.
     drawn: list[Any] = rng.choices(keys, cum_weights=cumulative, k=length)
     return drawn
 
 
 def _cumulative_table(elements: "OrderedDict[Any, float]") -> tuple[tuple[Any, ...], list[float]]:
-    table: tuple[tuple[Any, ...], list[float]] | None = getattr(elements, CACHE_ATTRIBUTE, None)
-    if table is None or len(table[0]) != len(elements):
-        table = (tuple(elements), list(accumulate(elements.values())))
-        setattr(elements, CACHE_ATTRIBUTE, table)
+    table = _cumulative.get(id(elements))
+    if table is None:
+        table = _cumulative[id(elements)] = (tuple(elements), list(accumulate(elements.values())))
     return table

{seedgraph-0.2.0 → seedgraph-0.2.1}/tests/test_sampling.py

@@ -5,6 +5,7 @@ from collections import OrderedDict
 import pytest
 from faker import Faker
 from faker.providers import BaseProvider
+from faker.providers.person import en_US
 from hypothesis import given
 from hypothesis import strategies as st
 from sqlalchemy import Column, ForeignKey, Integer, String, create_engine
@@ -65,16 +66,30 @@ def _pair(locale, seed_value):
     return untouched, seedgraph_fake
 
 
-def test_a_weighted_table_drawn_many_times_reads_its_weights_once():
-    table = CountingTable([("rare", 1), ("common", 8), ("medium", 3)])
+def test_a_table_faker_ships_reads_its_weights_once_however_many_draws(monkeypatch):
+    table = CountingTable(en_US.Provider.last_names)
+    monkeypatch.setattr(en_US.Provider, "last_names", table)
     fake = GenerationState().fake
 
     for _ in range(50):
-        fake.random_element(table)
+        fake.last_name()
 
     assert table.reads == 1
 
 
+def test_a_custom_table_changed_in_place_is_drawn_with_its_new_weights_like_faker_does():
+    table = OrderedDict([("before", 1), ("after", 0)])
+    untouched, seedgraph_fake = _pair(DEFAULT_LOCALE, 0)
+    draws = []
+    for fake in (untouched, seedgraph_fake):
+        table["before"], table["after"] = 1, 0
+        first = fake.random_element(table)
+        table["before"], table["after"] = 0, 1
+        draws.append((first, fake.random_element(table)))
+
+    assert draws[0] == draws[1] == ("before", "after")
+
+
 @given(
     locale=st.sampled_from(LOCALES),
     seed_value=st.integers(0, 2**32 - 1),

{seedgraph-0.2.0 → seedgraph-0.2.1}/uv.lock

@@ -1590,7 +1590,7 @@ wheels = [
 
 [[package]]
 name = "seedgraph"
-version = "0.2.0"
+version = "0.2.1"
 source = { editable = "." }
 dependencies = [
     { name = "faker" },