def test_is_not_square(self): for shape in [(3, 2), (2, 3)]: self.assertFalse(is_square(np.empty(shape)))
def test_is_square(self): self.assertTrue(is_square(np.empty((2, 2))))