|
1 | 1 | ''' |
2 | | -Pytest functions to test the AverageSweep experiment class |
| 2 | +Pytest functions to test the AverageExperiment experiment class |
3 | 3 | ''' |
4 | 4 |
|
5 | 5 |
|
@@ -101,16 +101,16 @@ def set_up_experiment(num_devices, measure_function, data_dir, verbose, n_averag |
101 | 101 | # instantiate expt based on additional parameters |
102 | 102 | if data_dir is None: |
103 | 103 | if verbose is False: |
104 | | - expt = ps.Sweep(runinfo, devices) |
| 104 | + expt = ps.Experiment(runinfo, devices) |
105 | 105 | elif verbose is True: |
106 | | - expt = ps.Sweep(runinfo, devices, verbose=verbose) |
| 106 | + expt = ps.Experiment(runinfo, devices, verbose=verbose) |
107 | 107 | else: |
108 | 108 | assert False, "Invalid verbose entry. Must be boolean." |
109 | 109 | elif isinstance(data_dir, str): |
110 | 110 | if verbose is False: |
111 | | - expt = ps.Sweep(runinfo, devices, data_dir) |
| 111 | + expt = ps.Experiment(runinfo, devices, data_dir) |
112 | 112 | elif verbose is True: |
113 | | - expt = ps.Sweep(runinfo, devices, data_dir, verbose) |
| 113 | + expt = ps.Experiment(runinfo, devices, data_dir, verbose) |
114 | 114 | else: |
115 | 115 | assert False, "Invalid verbose entry. Must be boolean." |
116 | 116 | else: |
@@ -184,9 +184,9 @@ def check_multi_data_results(expt, num_devices, shape1=[2], shape2=[2, 2], shape |
184 | 184 |
|
185 | 185 | ##################### TEST CASES BEGIN HERE ##################### |
186 | 186 |
|
187 | | -def test_averagesweep(): |
| 187 | +def test_averageexperiment(): |
188 | 188 | """ |
189 | | - Testing AverageSweep |
| 189 | + Testing AverageExperiment |
190 | 190 |
|
191 | 191 | Returns |
192 | 192 | -------- |
@@ -276,6 +276,6 @@ def test_variations(num_devices=0, measure_function=measure_point, data_dir=None |
276 | 276 | test_variations(bad=True) |
277 | 277 |
|
278 | 278 | 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" |
280 | 280 | 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