Writing a unittest with pytest

Question

I have a column in my Data-Frame and ]t should have a "Yes" values, But my Unit-Test code has an error. Here is my code:

data = func("value") assert all("Yes" in data == True)

The error is :

'bool' object is not iterable

What is "an error". Please paste the full contents of the error message. — blackbrandt
– blackbrandt, Commented Dec 31, 2021 at 16:17

OneCricketeer · Accepted Answer · 2021-12-31 17:15:39Z

If you simply want to check that one "Yes" string exists in the data variable

assert "Yes" in data

If you want to use check every element of the list is "Yes"

assert all(x == "Yes" for x in data)

However, this is for lists, not dataframes. If you have a pandas dataframe

assert data['column'].str.contains('Yes')

George Imerlishvili · Accepted Answer · 2021-12-31 17:08:01Z

0

all should take iterable as an argument not boolean that's why you get this error.

answered Dec 31, 2021 at 17:08

1,9852 gold badges16 silver badges24 bronze badges

2 Answers 2