def test_mcmc_run_1pole_cond(self):
     print(sys._getframe().f_code.co_name)
     test_iter = pc.piter(self.dataset1,
                          trial_run=False,
                          nsteps=2,
                          nwalkers=10,
                          number_of_poles=1,
                          nburn=1,
                          nthin=1)
     self.assertIsNotNone(test_iter.epsilon_iter)
     plt.close('all')
     return test_iter
 def test_mcmc_nrw_corr_run(self):
     print(sys._getframe().f_code.co_name)
     try:
         test_iter = pc.piter(self.dataset3,
                              trial_run=False,
                              nsteps=2,
                              nwalkers=15,
                              number_of_poles=0,
                              nburn=0,
                              nthin=1,
                              fit_mu=True,
                              number_of_poles_mu=0)
         return test_iter
     except Exception as e:
         raise
     plt.close('all')
 def test_waterpole(self):
     print(sys._getframe().f_code.co_name)
     try:
         test_iter = pc.piter(self.dataset2,
                              trial_run=False,
                              nsteps=5,
                              nwalkers=25,
                              number_of_poles=2,
                              nburn=1,
                              nthin=1,
                              fit_conductivity=True,
                              water_pole=True)
         return test_iter
     except Exception as e:
         raise
     plt.close('all')
 def test_run_nrw_corr_freq_cutoff(self):
     print(sys._getframe().f_code.co_name)
     test_iter = pc.piter(self.dataset3, start_freq=1e7, end_freq=3e9)
     self.assertIsNotNone(test_iter.meas)
     plt.close('all')
     return test_iter
 def test_run_nrw_corr(self):
     print(sys._getframe().f_code.co_name)
     test_iter = pc.piter(self.dataset3, fit_mu=True, number_of_poles_mu=0)
     self.assertIsNotNone(test_iter.meas)
     plt.close('all')
     return test_iter
 def test_run(self):
     print(sys._getframe().f_code.co_name)
     test_iter = pc.piter(self.dataset1)
     self.assertIsNotNone(test_iter.meas)
     plt.close('all')
     return test_iter