From 0f1df04f50c2ed789e2d77351497179b45b7588c Mon Sep 17 00:00:00 2001 From: Calum Chamberlain Date: Tue, 16 Jan 2024 14:29:59 +1300 Subject: [PATCH] Force Party instantiation to be made from Families I frequently accidentally instantiate a party with a string. This should not be allowed. --- eqcorrscan/core/match_filter/party.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/eqcorrscan/core/match_filter/party.py b/eqcorrscan/core/match_filter/party.py index c4e45407d..24e19430d 100644 --- a/eqcorrscan/core/match_filter/party.py +++ b/eqcorrscan/core/match_filter/party.py @@ -53,6 +53,10 @@ def __init__(self, families=None): if isinstance(families, Family): families = [families] if families: + assert isinstance(families, list), \ + "Families must be list of Family objects" + assert all(isinstance(f, Family) for f in families), \ + "Families must be list of Family objects" self.families.extend(families) def __repr__(self):