From 8dc331e7c28146e897ea3970b1c2a865b09b70ca Mon Sep 17 00:00:00 2001
From: Coda Phillips <cphillips@sean.ssec.wisc.edu>
Date: Mon, 25 Jul 2016 13:28:48 -0500
Subject: [PATCH] Check that all variables are updated when promised
---
util.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/util.py b/util.py
index 79d67bb..3414c71 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))
--
GitLab