assert inside if else condition in python

Question

I want to check the following using assert.

If: key mean is in dictionary d then it should have length equal to N. else: sigma should not be None.

What is the concise and pythonic way to write this assertion?

I tried this:

if "mean" in d:
    assert len(d["mean"]) == N
else:
    assert sigma is not None

Did you try any particular way of writing it? Did you find something unsatisfactory about your way of writing it? — Karl Knechtel
– Karl Knechtel, Commented Jul 5, 2022 at 22:43
@KarlKnechtel this is what I have in mind (updated the question) — melatonin15
– melatonin15, Commented Jul 5, 2022 at 22:49
This looks fine to me. You could reduce it to a single assert statement using a conditional expression to choose which comparison to assert, but it would be less readable. (assert len(...) == N if mean in d else sigma is not None) — chepner
– chepner, Commented Jul 5, 2022 at 22:52

Kache · Accepted Answer · 2022-07-06 03:34:38Z

1

A bit more code style thing, but if you want it to read like "it has one assert", I'd suggest:

has_expected_mean_len = 'mean' in d and len(d['mean']) == N
has_sigma = 'mean' not in d and sigma is not None

assert has_expected_mean_len or has_sigma

edited Jul 6, 2022 at 3:34

answered Jul 6, 2022 at 0:19

Kache

17k13 gold badges59 silver badges87 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

assert inside if else condition in python

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related