def test_get_valid_indices(self): """get_valid_indices should return correct result. """ control_ids = ['PC.354', 'PC.355', 'PC.356', 'PC.481', 'PC.593'] exp_control_indices = [0, 1, 2, 3, 4] fast_ids = ['PC.607', 'PC.634', 'PC.635', 'PC.636'] exp_fast_indices = [5, 6, 7, 8] obs_control = get_valid_indices(self.distance_header, control_ids) self.assertEqual(obs_control, exp_control_indices) obs_fast = get_valid_indices(self.distance_header, fast_ids) self.assertEqual(obs_fast, exp_fast_indices)
def test_get_valid_indices(self): """get_valid_indices should return correct result. """ control_ids = ['PC.354', 'PC.355', 'PC.356', 'PC.481', 'PC.593'] exp_control_indices = [0,1,2,3,4] fast_ids = ['PC.607', 'PC.634', 'PC.635', 'PC.636'] exp_fast_indices = [5,6,7,8] obs_control = get_valid_indices(self.distance_header, control_ids) self.assertEqual(obs_control, exp_control_indices) obs_fast = get_valid_indices(self.distance_header, fast_ids) self.assertEqual(obs_fast, exp_fast_indices)