[TASK] Use assertions in test and change spacing to follow style guide