[3/6] Overhaul tests; bundle C. elegans connectome for fixtures#7
Open
alexanderbates wants to merge 3 commits intoDrugowitschLab:mainfrom
Open
[3/6] Overhaul tests; bundle C. elegans connectome for fixtures#7alexanderbates wants to merge 3 commits intoDrugowitschLab:mainfrom
alexanderbates wants to merge 3 commits intoDrugowitschLab:mainfrom
Conversation
…as kwargs
Replaces the hardcoded NEG_NEUROTRANSMITTERS module constant with two
explicit constructor arguments so that the library no longer pre-empts
the user's neurotransmitter sign assignment:
- inhibitory_nts: pre-neuron top_nt values to negate when signed=True
(required when signed=True; raises ValueError otherwise).
- excluded_nts: pre-neuron top_nt values to drop entirely from W,
independent of signed=True/False. Useful for transmitter classes
whose net sign at a given target depends on the receptor mix and so
cannot be assigned a single sign safely.
Adds lambda_max as a constructor argument (default 0.99 for backwards
compatibility). _normalize_W now always rescales to lambda_max exactly
rather than only capping when the natural eigenvalue exceeds it, so the
parameter is a true control knob over leading-mode amplification rather
than just a stability ceiling. The amplification of the leading mode in
(I - W_rescaled)^-1 is 1 / (1 - lambda_max), so 0.99 gives ~100x and
0.5 gives ~2x.
Surfaces syn_weight_measure ('count' or 'norm') as a constructor
argument and changes the default from 'norm' to 'count'. Fixes a
pre-existing bug in _create_sparse_W: the signed=True path negated the
'count' column unconditionally, but the matrix was populated from the
column named by syn_weight_measure (default 'norm'), so the signed flag
silently produced the same matrix as signed=False. The negation now
applies to the column actually consumed. An inline comment notes that
flipping signs on 'norm' breaks the column-sums-to-1 interpretation, so
'count' is the more natural choice in signed mode.
Sign preservation: _build_influence_dataframe now keeps the real part
of the steady-state vector in signed mode rather than always taking the
magnitude, so net-inhibited targets carry a negative score.
Validates lambda_max in (0, 1) and syn_weight_measure in {'count',
'norm'}. When signed=True or excluded_nts is set, the SQLite meta
table must include a 'top_nt' column or _create_sparse_W raises.
calculate_influence now returns both the raw influence column and the three log-compressed adjusted_influence columns by default. Users can compare adjusted vs unadjusted scores from a single call rather than having to import adjust_influence and post-process the output themselves; opt out with adjust=False. The log compression is parameterised via two new kwargs on calculate_influence: adjust_const (the exp(-c) junk-node floor /+c shift, default 24) and adjust_signif (rounding, default 6). adjust_influence is added as a module-level function so advanced workflows can still post-process aggregated DataFrames (e.g. summing per-(target_class, seed_class) across multiple seeds before log compression in a worked example). Its output is three columns: - adjusted_influence = sign(x) * (log(max(|x|, exp(-const))) + const) - adjusted_influence_norm_by_targets (divides by n_targets per group) - adjusted_influence_norm_by_sources_and_targets (divides by n_sources * n_targets per group) The function dispatches on the presence of 'target' and 'seed' columns: when present it groups and sums per (target, seed); when absent it treats each row as its own group, which is the case for the DataFrame calculate_influence builds. Sign is preserved, so signed-mode input yields signed-mode output.
Replaces the legacy single-test scaffold (tests/test_InfluenceCalculator.py plus toy_network_example.sqlite and an example notebook) with a focused pytest suite that exercises the constructor surface introduced by the recent parameter rework and the integration of adjust_influence into calculate_influence: - Constructor validation: signed=True without inhibitory_nts raises; lambda_max outside (0, 1) raises (parametrised over five values); unknown syn_weight_measure raises. - Construction smoke: unsigned and signed builds, the 'norm' syn_weight_measure path, and excluded_nts dropping pre-neurons. Wrapped in pytest.importorskip so they skip cleanly on machines without PETSc/SLEPc rather than failing the suite. - adjust_influence helper: column shape, log-+-const anchor at the strongest influence, exp(-const) floor mapping zero raw influence to zero adjusted, sign preservation in signed mode, and the two validation errors (both score columns present, no score column). - calculate_influence integration: default returns adjusted columns alongside raw, adjust=False returns raw only, signed mode produces some net-negative downstream targets. - Bundled-data sanity check on column presence and row counts. Bundles the C. elegans hermaphrodite chemical connectome (300 cells, 3,539 edges, 20,672 synapses) under InfluenceCalculator/data/ as two CSVs plus an importable wrapper (celegans_edgelist(), celegans_meta()). Provenance and citation BibTeX live in the module docstring so help(InfluenceCalculator.data) surfaces them. The CSVs are an OpenWorm distribution extract (accessed February 2026) aggregating White et al. 1986 and Cook et al. 2019 with WormAtlas / CenGen annotations. The conftest fixture builds a temporary SQLite database from the bundled CSVs to drive InfluenceCalculator's still-SQLite-only constructor. Once the DataFrame / from_csv constructors land in a later PR the fixture can collapse to a path handoff.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Overhaul tests; bundle C. elegans connectome for fixtures
Replaces the legacy single-test scaffold (tests/test_InfluenceCalculator.py
plus toy_network_example.sqlite and an example notebook) with a focused
pytest suite that exercises the constructor surface introduced by the
recent parameter rework and the integration of adjust_influence into
calculate_influence:
lambda_max outside (0, 1) raises (parametrised over five values);
unknown syn_weight_measure raises.
syn_weight_measure path, and excluded_nts dropping pre-neurons.
Wrapped in pytest.importorskip so they skip cleanly on machines
without PETSc/SLEPc rather than failing the suite.
strongest influence, exp(-const) floor mapping zero raw influence to
zero adjusted, sign preservation in signed mode, and the two
validation errors (both score columns present, no score column).
alongside raw, adjust=False returns raw only, signed mode produces
some net-negative downstream targets.
Bundles the C. elegans hermaphrodite chemical connectome (300 cells,
3,539 edges, 20,672 synapses) under InfluenceCalculator/data/ as two
CSVs plus an importable wrapper (celegans_edgelist(),
celegans_meta()). Provenance and citation BibTeX live in the module
docstring so help(InfluenceCalculator.data) surfaces them. The CSVs
are an OpenWorm distribution extract (accessed February 2026)
aggregating White et al. 1986 and Cook et al. 2019 with WormAtlas /
CenGen annotations.
The conftest fixture builds a temporary SQLite database from the
bundled CSVs to drive InfluenceCalculator's still-SQLite-only
constructor. Once the DataFrame / from_csv constructors land in a
later PR the fixture can collapse to a path handoff.