@@ -381,8 +381,6 @@ def test_run_with_shots_option(self):
381
381
self .assertEqual (len (result ), 1 )
382
382
self .assertEqual (result [0 ].data .meas .num_shots , shots )
383
383
self .assertEqual (sum (result [0 ].data .meas .get_counts ().values ()), shots )
384
- self .assertIn ("shots" , result [0 ].metadata )
385
- self .assertEqual (result [0 ].metadata ["shots" ], shots )
386
384
387
385
with self .subTest ("default shots" ):
388
386
sampler = StatevectorSampler (seed = self ._seed )
@@ -391,8 +389,6 @@ def test_run_with_shots_option(self):
391
389
self .assertEqual (len (result ), 1 )
392
390
self .assertEqual (result [0 ].data .meas .num_shots , default_shots )
393
391
self .assertEqual (sum (result [0 ].data .meas .get_counts ().values ()), default_shots )
394
- self .assertIn ("shots" , result [0 ].metadata )
395
- self .assertEqual (result [0 ].metadata ["shots" ], default_shots )
396
392
397
393
with self .subTest ("setting default shots" ):
398
394
default_shots = 100
@@ -402,26 +398,20 @@ def test_run_with_shots_option(self):
402
398
self .assertEqual (len (result ), 1 )
403
399
self .assertEqual (result [0 ].data .meas .num_shots , default_shots )
404
400
self .assertEqual (sum (result [0 ].data .meas .get_counts ().values ()), default_shots )
405
- self .assertIn ("shots" , result [0 ].metadata )
406
- self .assertEqual (result [0 ].metadata ["shots" ], default_shots )
407
401
408
402
with self .subTest ("pub-like" ):
409
403
sampler = StatevectorSampler (seed = self ._seed )
410
404
result = sampler .run ([(bell , None , shots )]).result ()
411
405
self .assertEqual (len (result ), 1 )
412
406
self .assertEqual (result [0 ].data .meas .num_shots , shots )
413
407
self .assertEqual (sum (result [0 ].data .meas .get_counts ().values ()), shots )
414
- self .assertIn ("shots" , result [0 ].metadata )
415
- self .assertEqual (result [0 ].metadata ["shots" ], shots )
416
408
417
409
with self .subTest ("pub" ):
418
410
sampler = StatevectorSampler (seed = self ._seed )
419
411
result = sampler .run ([SamplerPub (bell , shots = shots )]).result ()
420
412
self .assertEqual (len (result ), 1 )
421
413
self .assertEqual (result [0 ].data .meas .num_shots , shots )
422
414
self .assertEqual (sum (result [0 ].data .meas .get_counts ().values ()), shots )
423
- self .assertIn ("shots" , result [0 ].metadata )
424
- self .assertEqual (result [0 ].metadata ["shots" ], shots )
425
415
426
416
with self .subTest ("multiple pubs" ):
427
417
sampler = StatevectorSampler (seed = self ._seed )
@@ -437,13 +427,9 @@ def test_run_with_shots_option(self):
437
427
self .assertEqual (len (result ), 2 )
438
428
self .assertEqual (result [0 ].data .meas .num_shots , shots1 )
439
429
self .assertEqual (sum (result [0 ].data .meas .get_counts ().values ()), shots1 )
440
- self .assertIn ("shots" , result [0 ].metadata )
441
- self .assertEqual (result [0 ].metadata ["shots" ], shots1 )
442
430
443
431
self .assertEqual (result [1 ].data .meas .num_shots , shots2 )
444
432
self .assertEqual (sum (result [1 ].data .meas .get_counts ().values ()), shots2 )
445
- self .assertIn ("shots" , result [1 ].metadata )
446
- self .assertEqual (result [1 ].metadata ["shots" ], shots2 )
447
433
448
434
def test_run_shots_result_size (self ):
449
435
"""test with shots option to validate the result size"""
@@ -647,8 +633,8 @@ def test_metadata(self):
647
633
648
634
self .assertEqual (len (result ), 2 )
649
635
self .assertIsInstance (result .metadata , dict )
650
- self .assertEqual (result [0 ].metadata , {"shots" : 10 , " circuit_metadata" : qc .metadata })
651
- self .assertEqual (result [1 ].metadata , {"shots" : 10 , " circuit_metadata" : qc2 .metadata })
636
+ self .assertEqual (result [0 ].metadata , {"circuit_metadata" : qc .metadata })
637
+ self .assertEqual (result [1 ].metadata , {"circuit_metadata" : qc2 .metadata })
652
638
653
639
654
640
if __name__ == "__main__" :
0 commit comments