diff --git a/util.py b/util.py index 79d67bbe700441977c02206d5e00d3e72f7b7fbd..3414c713be6413e7cd9c39c9c5895bc77094cf04 100644 --- a/util.py +++ b/util.py @@ -120,8 +120,9 @@ class BaseCheckList: def compute(self, frame): # Filter bad records from previous level filtered_frame = frame.ix[frame.qc_percent < 1].copy() - for check in self.checks: + for check, metadata in self.checks.items(): original_shape = filtered_frame.shape filtered_frame = check(filtered_frame, self.parameters) assert filtered_frame.shape[0] == original_shape[0] + assert np.in1d(list(metadata['updates']), filtered_frame.columns).all() return self.update_qc_percent(filtered_frame.combine_first(frame))