Skip to content

Commit 7f76861

Browse files
rsbrosti-am-mounce
authored andcommitted
chore: fixing error causing build and tests to fail.
1 parent 95c577a commit 7f76861

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

test/measurement/test_average_experiment.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
'''
2-
Pytest functions to test the AverageSweep experiment class
2+
Pytest functions to test the AverageExperiment experiment class
33
'''
44

55

@@ -101,16 +101,16 @@ def set_up_experiment(num_devices, measure_function, data_dir, verbose, n_averag
101101
# instantiate expt based on additional parameters
102102
if data_dir is None:
103103
if verbose is False:
104-
expt = ps.Sweep(runinfo, devices)
104+
expt = ps.Experiment(runinfo, devices)
105105
elif verbose is True:
106-
expt = ps.Sweep(runinfo, devices, verbose=verbose)
106+
expt = ps.Experiment(runinfo, devices, verbose=verbose)
107107
else:
108108
assert False, "Invalid verbose entry. Must be boolean."
109109
elif isinstance(data_dir, str):
110110
if verbose is False:
111-
expt = ps.Sweep(runinfo, devices, data_dir)
111+
expt = ps.Experiment(runinfo, devices, data_dir)
112112
elif verbose is True:
113-
expt = ps.Sweep(runinfo, devices, data_dir, verbose)
113+
expt = ps.Experiment(runinfo, devices, data_dir, verbose)
114114
else:
115115
assert False, "Invalid verbose entry. Must be boolean."
116116
else:
@@ -184,9 +184,9 @@ def check_multi_data_results(expt, num_devices, shape1=[2], shape2=[2, 2], shape
184184

185185
##################### TEST CASES BEGIN HERE #####################
186186

187-
def test_averagesweep():
187+
def test_averageexperiment():
188188
"""
189-
Testing AverageSweep
189+
Testing AverageExperiment
190190
191191
Returns
192192
--------
@@ -276,6 +276,6 @@ def test_variations(num_devices=0, measure_function=measure_point, data_dir=None
276276
test_variations(bad=True)
277277

278278
with pytest.raises(Exception):
279-
test_variations(n_average=-1), "Averagesweep's n_average must be 1 or more"
279+
test_variations(n_average=-1), "AverageExperiment's n_average must be 1 or more"
280280
with pytest.raises(Exception):
281-
test_variations(n_average=0), "Averagesweep's n_average must be 1 or more"
281+
test_variations(n_average=0), "AverageExperiment's n_average must be 1 or more"

0 commit comments

Comments
 (0)