예제 #1
0
 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)
예제 #2
0
 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)
예제 #3
0
 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])
예제 #4
0
    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)))
예제 #5
0
 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])
예제 #6
0
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)