diff options
| -rw-r--r-- | archaeological_context_records/tests.py | 2 | ||||
| -rw-r--r-- | archaeological_operations/tests.py | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py index 8ce2fd9d5..a242799ef 100644 --- a/archaeological_context_records/tests.py +++ b/archaeological_context_records/tests.py @@ -108,6 +108,8 @@ class ContextRecordInit(OperationInitTest): return self.context_records def get_default_context_record(self, force=False): + if force: + return self.create_context_record(force=force)[-1] return self.create_context_record(force=force)[0] def tearDown(self): diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py index 93c4fbe92..7d11ee002 100644 --- a/archaeological_operations/tests.py +++ b/archaeological_operations/tests.py @@ -496,6 +496,8 @@ class OperationInitTest(object): return self.operations def get_default_operation(self, force=False): + if force: + return self.create_operation()[-1] return self.create_operation()[0] def tearDown(self): |