MCPcopy Index your code
hub / github.com/pytorch/pytorch / testPreservesMetadata

Method testPreservesMetadata

caffe2/python/schema_test.py:185–222  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

183 s['fields2:keys:non_existent']
184
185 def testPreservesMetadata(self):
186 s = schema.Struct(
187 ('a', schema.Scalar(np.float32)), (
188 'b', schema.Scalar(
189 np.int32,
190 metadata=schema.Metadata(categorical_limit=5)
191 )
192 ), (
193 'c', schema.List(
194 schema.Scalar(
195 np.int32,
196 metadata=schema.Metadata(categorical_limit=6)
197 )
198 )
199 )
200 )
201 # attach metadata to lengths field
202 s.c.lengths.set_metadata(schema.Metadata(categorical_limit=7))
203
204 self.assertEqual(None, s.a.metadata)
205 self.assertEqual(5, s.b.metadata.categorical_limit)
206 self.assertEqual(6, s.c.value.metadata.categorical_limit)
207 self.assertEqual(7, s.c.lengths.metadata.categorical_limit)
208 sc = s.clone()
209 self.assertEqual(None, sc.a.metadata)
210 self.assertEqual(5, sc.b.metadata.categorical_limit)
211 self.assertEqual(6, sc.c.value.metadata.categorical_limit)
212 self.assertEqual(7, sc.c.lengths.metadata.categorical_limit)
213 sv = schema.from_blob_list(
214 s, [
215 np.array([3.4]), np.array([2]), np.array([3]),
216 np.array([1, 2, 3])
217 ]
218 )
219 self.assertEqual(None, sv.a.metadata)
220 self.assertEqual(5, sv.b.metadata.categorical_limit)
221 self.assertEqual(6, sv.c.value.metadata.categorical_limit)
222 self.assertEqual(7, sv.c.lengths.metadata.categorical_limit)
223
224 def testDupField(self):
225 with self.assertRaises(ValueError):

Callers

nothing calls this directly

Calls 3

cloneMethod · 0.95
set_metadataMethod · 0.45
assertEqualMethod · 0.45

Tested by

no test coverage detected