Write test for rectangle