From 25b3453919345c7c9a93d291e78afc006cd984f4 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Tue, 20 Dec 2016 18:23:34 +0100 Subject: Fix default test --- archaeological_context_records/tests.py | 2 ++ archaeological_operations/tests.py | 2 ++ 2 files changed, 4 insertions(+) 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): -- cgit v1.2.3