def test_disable_prints(self): conf.set_verbose(0) tmp = conf.CDDMConfig["verbose"] out = pt.disable_prints() self.assertEqual(out, tmp) self.assertEqual(0, conf.CDDMConfig["verbose"]) pt.enable_prints(out) self.assertEqual(tmp, conf.CDDMConfig["verbose"])
def test_set_verbose(self): conf.set_verbose(0) self.assertEqual(0, conf.set_verbose(1)) self.assertEqual(1, conf.set_verbose(2)) self.assertEqual(2, conf.set_verbose(0)) with self.assertRaises(ValueError): conf.set_verbose("a")
def test_print(self): conf.set_verbose(0) pt.print1("No print") pt.print2("No print") pt.print_progress(4, 10) pt.print_progress(10, 10) conf.set_verbose(1) pt.print1("Ok") pt.print2("No print") conf.set_verbose(2) pt.print1("OK") pt.print2("OK") pt.print_progress(4, 10) pt.print_progress(10, 10)
""" In this example we use a simulated single-camera video of o brownian motion of spherical particles and perform ddm analysis. You must first create FFTs of the videos by calling simple_brownian_fft.py """ from cddm import normalize, k_select, adiff import matplotlib.pyplot as plt from cddm import conf import numpy as np #setting this to 2 shows progress bar conf.set_verbose(2) SHAPE = (512, 512) v = np.load("simple_brownian_ddm_fft.npy") nframes = len(v) data, count = adiff(v, n=2**10) data = normalize((data, count)) i, j = 4, 8 plt.figure()
def test_frame_rate(self): import time t0 = time.time() - 1. conf.set_verbose(2) pt.print_frame_rate(1024, t0) pt.print_frame_rate(1024, t0, time.time())