Python PythonTools.Debug类代码示例

def make_ft_axis(length, dt, undersampling = 0, normalized_to_period = 0, zero_in_middle = False, flag_verbose = False):
    """
    fourier transform the time axis
    20101204/RB: started
    20130131/RB: now uses speed of light value from Crocodile.Resources.Constants

    INPUT:
    length: amount of samples
    dt: time between samples
    undersampling: 
    normalized_to_period: will normalize the spectrum to 1 for the specified period. If the number is 0, the spectrum will not be normalized.
    zero_in_middle: determines the frequency axes.

    OUTPUT:
    A frequency axis.     

    """
    DEBUG.verbose("make FT axis", flag_verbose)
    if normalized_to_period == 0:   
        resolution = 1 / ( CONST.wavenumberToInvFs * length * dt)
    else:
        resolution = normalized_to_period / (length * dt)

    array = numpy.arange((undersampling)*length/2, (undersampling+1)*length/2)*resolution

    if zero_in_middle == False:
        return numpy.concatenate((array,-numpy.flipud(array)))
    else:
        return numpy.concatenate((-numpy.flipud(array), array))

def execute(args):

    if args.skip1 == False:
        suite = unittest.TestLoader().loadTestsFromTestCase(Test_setting_file_stuff)
        unittest.TextTestRunner(verbosity=1).run(suite) 
    else:
        DEBUG.verbose("Skipping: " + suite_list[0], True)

def execute(args):

    if args.skip1 == False:
        suite = unittest.TestLoader().loadTestsFromTestCase(Test_find_axes)
        unittest.TextTestRunner(verbosity=1).run(suite)  
    else:
        DEBUG.verbose("Skipping: " + suite_list[0], True)


    if args.skip2 == False:
        suite = unittest.TestLoader().loadTestsFromTestCase(Test_make_contours_2d)
        unittest.TextTestRunner(verbosity=1).run(suite)    
    else:
        DEBUG.verbose("Skipping: " + suite_list[1], True)   

    if args.skip3 == False:
        suite = unittest.TestLoader().loadTestsFromTestCase(Test_find_axes_indices)
        unittest.TextTestRunner(verbosity=1).run(suite)    
    else:
        DEBUG.verbose("Skipping: " + suite_list[2], True)        

    if args.skip4 == False:
        suite = unittest.TestLoader().loadTestsFromTestCase(Test_truncate_data)
        unittest.TextTestRunner(verbosity=1).run(suite)    
    else:
        DEBUG.verbose("Skipping: " + suite_list[3], True)  

    if args.skip5 == False:
        suite = unittest.TestLoader().loadTestsFromTestCase(Test_find_subplots)
        unittest.TextTestRunner(verbosity=1).run(suite)    
    else:
        DEBUG.verbose("Skipping: " + suite_list[4], True)  

def truncate_data(data, x_axis, y_axis, x_min_i, x_max_i, y_min_i, y_max_i, flag_verbose = False):
    """
    Truncate data in a non-default way. 
    x_axis = [1,2,3,4,5]
    x_min_i = 0
    if x_max_i == 2: x_axis = [1,2]
    if x_max_i == 4: x_axis = [1,2,3,4]
    if x_max_i == -1: x_axis = [1,2,3,4,5]
    
    CHANGELOG:
    20130213: was in Plotting.contourplot function, moved to Functions module, now handles x_max_i == -1
    """
    
    DEBUG.verbose("Truncate data", flag_verbose)

    if x_max_i == -1:
        DEBUG.verbose("  x_max == -1", flag_verbose)
        data = data[:,x_min_i:]
        x_axis = x_axis[x_min_i:]
    else:
        DEBUG.verbose("  x_max != -1", flag_verbose)
        data = data[:,x_min_i:x_max_i]
        x_axis = x_axis[x_min_i:x_max_i]        

    if y_max_i == -1:
        DEBUG.verbose("  y_max == -1", flag_verbose)
        data = data[y_min_i:,:]
        y_axis = y_axis[y_min_i:]   
    else:
        DEBUG.verbose("  y_max != -1", flag_verbose)
        data = data[y_min_i:y_max_i,:]
        y_axis = y_axis[y_min_i:y_max_i]         
        
    return data, x_axis, y_axis

 def test_time_stamp_False(self):
     """
     This is correct
     """
     DEBUG.verbose("\nWarning is intentional", True)
     self.dc.time_stamp = False
     self.assertEqual(self.dc.time_stamp, "False")

 def test_time_stamp_5_int(self):
     """
     This is correct
     """
     DEBUG.verbose("\nWarning is intentional", True)
     self.dc.time_stamp = 12345
     self.assertEqual(self.dc.time_stamp, "12345")

 def test_import_data_no_data(self):
     """
     folder exists, LV_file_format.1 exists and is correct, but contains no other data
     """
     self.mess.path = "Test_resources/petw_test_folder_2/"
     DEBUG.verbose("\nError about importing is intentional", True)
     res = self.mess.import_data(flag_verbose = self.flag_verbose)
     self.assertFalse(res)

 def test_phase_rad_nan(self):
     """
     None sets phase_degrees to init value
     """
     DEBUG.verbose("\nWarning is intentional", True)
     self.dc.phase_rad = None
     self.assertEqual(self.dc.phase_degrees, None)
     self.assertEqual(self.dc.phase_rad, None)

 def test_2p_1m(self):
     """
     mess[0].s = 1 + [1].s: 2 - [2].s: 3
     (1 + 2) / 2 - 3 = -1.5
     """
     DEBUG.verbose("\nWarning about zeropad intentional", True)
     mer = PEME.pe_merge("Test", class_plus = [self.mess[0], self.mess[1]], class_min = [self.mess[2]], flag_verbose = self.flag_verbose)
     self.assertTrue(numpy.all(mer.s == -1.5))

 def test_phase_rad_true(self):
     """
     True should be interpreted as 1
     """
     DEBUG.verbose("\nWarning is intentional", True)
     self.dc.phase_rad = True
     self.assertAlmostEqual(self.dc.phase_degrees, 1 * 180 / numpy.pi)
     self.assertEqual(self.dc.phase_rad, 1)

 def test_phase_rad_false(self):
     """
     False should be interpreted as 0
     """
     DEBUG.verbose("\nWarning is intentional", True)
     self.dc.phase_rad = False
     self.assertEqual(self.dc.phase_degrees, 0)
     self.assertEqual(self.dc.phase_rad, 0)

 def test_phase_degrees_none(self):
     """
     None should will reset phase_degrees to init value
     """
     DEBUG.verbose("\nWarning is intentional", True)
     self.dc.phase_degrees = None
     self.assertEqual(self.dc.phase_degrees, None)
     self.assertEqual(self.dc.phase_rad, None)

 def test_import_data_incorrect_path(self):
     """
     Path is incorrect, no data imported. Returns False
     """
     self.mess.path = "Test_resources/FolderDoesNotExist/"
     DEBUG.verbose("\nError that directory can't found is intentional", True)
     res = self.mess.import_data(flag_verbose = self.flag_verbose)
     self.assertFalse(res)

 def test_zeropad_to_r_uninit(self):
     """
     without r set, it should give an error
     """
     DEBUG.verbose("\nError is intentional", True)
     self.dc.zeropad_to = 40
     self.assertEqual(self.dc.zeropad_to, None)
     self.assertEqual(self.dc.zeropad_by, 1.0)

 def test_import_data_wrong_file_format(self):
     """
     LV_file_format.666 should not be recognized.
     """
     self.mess.path = "Test_resources/petw_test_folder_1/"
     DEBUG.verbose("\nError about unknown file format is intentional", True)
     res = self.mess.import_data(flag_verbose = self.flag_verbose)
     self.assertFalse(res)