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())
