| from __future__ import annotations |
|
|
| import numpy as np |
|
|
| from toaster.core import Grouping |
|
|
|
|
| def test_groups_exclude_noise(): |
| g = Grouping(np.array([0, 0, 1, 1, 1, -1], np.int32)) |
| assert g.n_groups == 2 |
| assert g.group_ids().tolist() == [0, 1] |
| assert g.n == 6 |
|
|
|
|
| def test_indices_of_and_group_of(): |
| g = Grouping(np.array([2, 0, 2, -1, 0], np.int32)) |
| assert g.indices_of(2).tolist() == [0, 2] |
| assert g.indices_of(0).tolist() == [1, 4] |
| assert g.group_of(3) == -1 |
| |
| assert g.indices_of(99).tolist() == [] |
|
|
|
|
| def test_metadata_defaults(): |
| g = Grouping(np.array([0, 1], np.int32), source="dbscan", params={"eps": 0.5}) |
| assert g.source == "dbscan" |
| assert g.params["eps"] == 0.5 |
| assert g.suggested_labels is None |
|
|
