def runTest(self): d1 = dum_mats.row_offset_col_gain_posNegImag() sorted_keys = d1.sorted_keys() sorted_val = d1.sorted_values() for i, val in enumerate(sorted_keys): i_off = i - 10 self.check_pos_neg_imag_sub(d1, sorted_val, i, i_off, val)
def runTest(self): d1 = dum_mats.row_offset_col_gain_posNegImag() d2 = d1[1:-1:2] self.assertEqual(len(d2), 10) sorted_keys = d2.sorted_keys() sorted_val = d2.sorted_values() for i, val in enumerate(sorted_keys): i_off = i * 2 - 9 self.check_pos_neg_imag_sub(d1, sorted_val, i, i_off, val)
def runTest(self): d1 = dum_mats.row_offset_col_gain_posNegImag() self.assertEqual(len(d1), 21) sorted_keys = d1.sorted_keys() sorted_val = d1.sorted_values() for i in range(21): kv = d1[i] self.assertEqual(kv[0], sorted_keys[i]) nw.are_matrices_close(kv[1], sorted_val[i])
def runTest(self): d1 = dum_mats.row_offset_col_gain_posNegImag() ret = d1.get_slice_indices(0, 20, 3) self.assertEqual(ret[0], (0, 21, 10)) self.assertEqual( ret[1], (self.cal_pos_neg_imag_ene(-10), self.cal_pos_neg_imag_ene(10))) ret = d1.get_slice_indices(1, 19, 3) self.assertEqual(ret[0], (1, 20, 9)) self.assertEqual( ret[1], (self.cal_pos_neg_imag_ene(-9), self.cal_pos_neg_imag_ene(9))) ret = d1.get_slice_indices(0, 20, 4) self.assertEqual(ret[0], (0, 19, 6)) self.assertEqual( ret[1], (self.cal_pos_neg_imag_ene(-10), self.cal_pos_neg_imag_ene(8))) ret = d1.get_slice_indices(1, 19, 4) self.assertEqual(ret[0], (1, 20, 6)) self.assertEqual( ret[1], (self.cal_pos_neg_imag_ene(-9), self.cal_pos_neg_imag_ene(9))) ret = d1.get_slice_indices(0, 20, 10) self.assertEqual(ret[0], (0, 19, 2)) self.assertEqual( ret[1], (self.cal_pos_neg_imag_ene(-10), self.cal_pos_neg_imag_ene(8))) ret = d1.get_slice_indices(1, 19, 10) self.assertEqual(ret[0], (1, 20, 2)) self.assertEqual( ret[1], (self.cal_pos_neg_imag_ene(-9), self.cal_pos_neg_imag_ene(9))) ret = d1.get_slice_indices(2, 19, 10) self.assertEqual(ret[0], (2, 12, 1)) self.assertEqual( ret[1], (self.cal_pos_neg_imag_ene(-8), self.cal_pos_neg_imag_ene(1)))
def runTest(self): d_ryd = dum_mats.row_offset_col_gain_posNegImag() d_eV = d_ryd.convert_ene_units(tu.eVs) for val in zip(d_ryd.sorted_keys(), d_eV.sorted_keys()): self.assertAlmostEqual(val[0] * tu.rydbergs_to_eVs, val[1])
import os import sys from __builtin__ import raw_input basedir = os.path.dirname(os.path.realpath(__file__)) sys.path.insert(0, basedir + '/../..') import dum_mats d = dum_mats.row_offset_col_gain_zeroimag() print str(d) raw_input("Any key to continue.") d = dum_mats.row_offset_col_gain_posImag() print str(d) raw_input("Any key to continue.") d = dum_mats.row_offset_col_gain_negImag() print str(d) raw_input("Any key to continue.") d = dum_mats.row_offset_col_gain_posNegImag() print str(d) raw_input("Any key to continue.") d = d.create_reduced_dim(0) print str(d) raw_input("Any key to continue.") d = d.create_reduced_dim(0) print str(d)