Address review comment: Badly defined test data.
Address review comment: More accurate docstring.
Address review comment: Fix formatting.
Address review comment: More accurate test.
Address review comment: Simplify.