|  |  | @ -6,7 +6,7 @@ import numpy as np | 
			
		
	
		
		
			
				
					
					|  |  |  | 
 |  |  |  | 
 | 
			
		
	
		
		
			
				
					
					|  |  |  | import cereal.messaging as messaging |  |  |  | import cereal.messaging as messaging | 
			
		
	
		
		
			
				
					
					|  |  |  | from common.params import Params |  |  |  | from common.params import Params | 
			
		
	
		
		
			
				
					
					|  |  |  | from selfdrive.locationd.calibrationd import Calibrator, INPUTS_NEEDED, INPUTS_WANTED, BLOCK_SIZE, MIN_SPEED_FILTER, MAX_YAW_RATE_FILTER |  |  |  | from selfdrive.locationd.calibrationd import Calibrator, INPUTS_NEEDED, INPUTS_WANTED, BLOCK_SIZE, MIN_SPEED_FILTER, MAX_YAW_RATE_FILTER, SMOOTH_CYCLES | 
			
				
				
			
		
	
		
		
	
		
		
			
				
					
					|  |  |  | 
 |  |  |  | 
 | 
			
		
	
		
		
			
				
					
					|  |  |  | 
 |  |  |  | 
 | 
			
		
	
		
		
			
				
					
					|  |  |  | class TestCalibrationd(unittest.TestCase): |  |  |  | class TestCalibrationd(unittest.TestCase): | 
			
		
	
	
		
		
			
				
					|  |  | @ -86,7 +86,10 @@ class TestCalibrationd(unittest.TestCase): | 
			
		
	
		
		
			
				
					
					|  |  |  |                          [1e-3, 1e-3, 1e-3]) |  |  |  |                          [1e-3, 1e-3, 1e-3]) | 
			
		
	
		
		
			
				
					
					|  |  |  |     self.assertEqual(c.valid_blocks, INPUTS_WANTED) |  |  |  |     self.assertEqual(c.valid_blocks, INPUTS_WANTED) | 
			
		
	
		
		
			
				
					
					|  |  |  |     np.testing.assert_allclose(c.rpy, [0.0, 0.0, 0.0]) |  |  |  |     np.testing.assert_allclose(c.rpy, [0.0, 0.0, 0.0]) | 
			
		
	
		
		
			
				
					
					|  |  |  |     for _ in range(BLOCK_SIZE): |  |  |  |     old_rpy_weight_prev = 0.0 | 
			
				
				
			
		
	
		
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  |     for _ in range(BLOCK_SIZE + 10): | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  |       self.assertLess(old_rpy_weight_prev - c.old_rpy_weight, 1/SMOOTH_CYCLES + 1e-3) | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  |       old_rpy_weight_prev = c.old_rpy_weight | 
			
		
	
		
		
			
				
					
					|  |  |  |       c.handle_v_ego(MIN_SPEED_FILTER + 1) |  |  |  |       c.handle_v_ego(MIN_SPEED_FILTER + 1) | 
			
		
	
		
		
			
				
					
					|  |  |  |       c.handle_cam_odom([MIN_SPEED_FILTER + 1, -0.05 * MIN_SPEED_FILTER, 0.0], |  |  |  |       c.handle_cam_odom([MIN_SPEED_FILTER + 1, -0.05 * MIN_SPEED_FILTER, 0.0], | 
			
		
	
		
		
			
				
					
					|  |  |  |                          [0.0, 0.0, 0.0], |  |  |  |                          [0.0, 0.0, 0.0], | 
			
		
	
	
		
		
			
				
					|  |  | 
 |