Skip to content

Commit 9e51d66

Browse files
Jammy2211Jammy2211
authored andcommitted
fix test_autofit/non_linear/search/test_abstract_search.py
1 parent 76110b2 commit 9e51d66

2 files changed

Lines changed: 15 additions & 2 deletions

File tree

autofit/non_linear/mock/mock_result.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,13 @@ def __init__(
3838
self.search = search
3939
self.model = model
4040

41+
@property
42+
def model_centred(self):
43+
try:
44+
return self.samples_summary.model_centred
45+
except AttributeError:
46+
return self.model
47+
4148
def model_centred_absolute(self, a):
4249
try:
4350
return self.samples_summary.model_centred_absolute(a)

test_autofit/non_linear/search/test_abstract_search.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,19 +48,25 @@ class TestResult:
4848
def test_model(self, result):
4949

5050
component = result.model.component
51+
assert component.one_tuple.one_tuple_0.mean == 0.5
52+
assert component.one_tuple.one_tuple_1.mean == 1
53+
54+
def test_model_centred(self, result):
55+
56+
component = result.model_centred.component
5157
assert component.one_tuple.one_tuple_0.mean == 0
5258
assert component.one_tuple.one_tuple_1.mean == 1
5359
assert component.one_tuple.one_tuple_0.sigma == 0.2
5460
assert component.one_tuple.one_tuple_1.sigma == 0.2
5561

56-
def test_model_absolute(self, result):
62+
def test_model_centred_absolute(self, result):
5763
component = result.model_centred_absolute(a=2.0).component
5864
assert component.one_tuple.one_tuple_0.mean == 0
5965
assert component.one_tuple.one_tuple_1.mean == 1
6066
assert component.one_tuple.one_tuple_0.sigma == 2.0
6167
assert component.one_tuple.one_tuple_1.sigma == 2.0
6268

63-
def test_model_relative(self, result):
69+
def test_model_centred_relative(self, result):
6470
component = result.model_centred_relative(r=1.0).component
6571
assert component.one_tuple.one_tuple_0.mean == 0
6672
assert component.one_tuple.one_tuple_1.mean == 1

0 commit comments

Comments
 (0)