The assert for answer 12.2.2. seem to check the dimension of the output DataFrame: (20, 3). For checking the number of columns it asserts all(len(i) == 3 for i in output), this races an unspecified error. Wouldn't len(output.columns) == 3 be more a robust assertion?
The assert for answer 12.2.2. seem to check the dimension of the output DataFrame: (20, 3). For checking the number of columns it asserts
all(len(i) == 3 for i in output)
, this races an unspecified error. Wouldn'tlen(output.columns) == 3
be more a robust assertion?