Skip to content

Commit

Permalink
added a test for adaptive sampling
Browse files Browse the repository at this point in the history
  • Loading branch information
AdriaanRol committed May 17, 2018
1 parent 68ae7f7 commit eb27f09
Showing 1 changed file with 16 additions and 0 deletions.
16 changes: 16 additions & 0 deletions pycqed/tests/test_MeasurementControl.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
import pycqed as pq
import unittest
import numpy as np
import adaptive
import pycqed.analysis.analysis_toolbox as a_tools
from pycqed.measurement import measurement_control
from pycqed.measurement.sweep_functions import None_Sweep, None_Sweep_idx
Expand Down Expand Up @@ -555,6 +556,21 @@ def test_adaptive_measurement_SPSA(self):
self.assertLess(yf, 0.7)
self.assertLess(pf, 0.7)


def test_adaptive_sampling(self):
self.MC.soft_avg(1)
self.mock_parabola.noise(0)
self.MC.set_sweep_functions(
[self.mock_parabola.x, self.mock_parabola.y])
self.MC.set_adaptive_function_parameters({'adaptive_function': adaptive.Learner2D,
'goal':lambda l: l.npoints>20*20,
'bounds':((-50, +50),
(-20, +30))})
self.MC.set_detector_function(self.mock_parabola.parabola)
dat = self.MC.run('2D adaptive sampling test', mode='adaptive')



@classmethod
def tearDownClass(self):
self.MC.close()
Expand Down

0 comments on commit eb27f09

Please sign in to comment.