def test_reshape_view_with_negative_value_has_correct_shape(self): b = Matrix(12, 2, 1) self.assertEqual(b.reshape(-1, 3, 2).shape(), (4, 3, 2)) self.assertEqual(b.reshape(1, 1, -1).shape(), (1, 1, 24)) self.assertEqual(b.reshape(1, -1, 4).shape(), (1, 6, 4))
def test_reshape_view_has_correct_shape(self): b = Matrix(12, 2, 1) self.assertEqual(b.reshape(4, 3, 2).shape(), (4, 3, 2)) self.assertEqual(b.reshape(1, 1, 24).shape(), (1, 1, 24)) self.assertEqual(b.reshape(1, 6, 4).shape(), (1, 6, 4))