@@ -94,6 +94,7 @@ def test_run_poly_example_with_design_matrix_and_genkw_merge(default_values):
9494 "REAL" : list (range (num_realizations )),
9595 "a" : a_values ,
9696 "category" : 5 * ["cat1" ] + 5 * ["cat2" ],
97+ "big_integer" : [1e10 + i for i in range (num_realizations )],
9798 }
9899 ),
99100 pl .DataFrame (default_values , orient = "row" ),
@@ -117,10 +118,11 @@ def test_run_poly_example_with_design_matrix_and_genkw_merge(default_values):
117118 encoding = "utf-8" ,
118119 )
119120
120- # This adds a dummy category parameter to COEFFS GENKW
121- # which will be overridden by the design matrix catagorical entries
121+ # This adds a dummy category and big_numbers parameter to COEFFS GENKW
122+ # which will be overridden by the design matrix entries
122123 with open ("coeff_priors" , "a" , encoding = "utf-8" ) as f :
123- f .write ("category UNIFORM 0 1" )
124+ f .write ("category UNIFORM 0 1\n " )
125+ f .write ("big_numbers UNIFORM 0 1\n " )
124126
125127 Path ("poly_eval.py" ).write_text (
126128 dedent (
@@ -153,6 +155,7 @@ def _evaluate(coeffs, x):
153155 b: <b>
154156 c: <c>
155157 category: <category>
158+ big_integer: <big_integer>
156159 """
157160 ),
158161 encoding = "utf-8" ,
@@ -185,6 +188,9 @@ def _evaluate(coeffs, x):
185188 np .testing .assert_array_equal (
186189 params ["category" ].to_list (), 5 * ["cat1" ] + 5 * ["cat2" ]
187190 )
191+ np .testing .assert_array_equal (
192+ params ["big_integer" ].to_list (), [1e10 + i for i in range (10 )]
193+ )
188194 np .testing .assert_array_equal (params ["b" ].to_list (), 10 * [1 ])
189195 np .testing .assert_array_equal (params ["c" ].to_list (), 10 * [2 ])
190196 with open ("poly_out/realization-0/iter-0/my_output" , encoding = "utf-8" ) as f :
@@ -193,12 +199,14 @@ def _evaluate(coeffs, x):
193199 assert output [1 ] == "b: 1"
194200 assert output [2 ] == "c: 2"
195201 assert output [3 ] == "category: cat1"
202+ assert output [4 ] == "big_integer: 10000000000"
196203 with open ("poly_out/realization-5/iter-0/my_output" , encoding = "utf-8" ) as f :
197204 output = [line .strip () for line in f ]
198205 assert output [0 ] == "a: 5"
199206 assert output [1 ] == "b: 1"
200207 assert output [2 ] == "c: 2"
201208 assert output [3 ] == "category: cat2"
209+ assert output [4 ] == "big_integer: 10000000005"
202210
203211
204212@pytest .mark .usefixtures (
0 commit comments