refractored tests to expect(). syntax