def test_construction_with_shape(self): b = Matrix(2, 3, 4) self.assertEqual(len(b), 2 * 3 * 4) self.assertEqual(b.get_feature_size(), 4) self.assertEqual(b.get_batch_size(), 3) self.assertEqual(b.get_time_size(), 2) self.assertEqual(b.shape(), (2, 3, 4))