Skip to content

Commit 0e8f77d

Browse files
fixed constraint unit test
1 parent 2a24bd9 commit 0e8f77d

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

batchglm/unit_test/test_acc_constrained_vglm_all.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,11 @@ def simulate(self):
2424
constraints[1, 1] = 1
2525
constraints[2, 2] = 1
2626
constraints[3, 2] = -1
27+
new_coef_names = ['Intercept', 'condition[T.1]', 'batch[1]', 'batch[2]']
2728
self.sim1.input_data.design_loc = dmat
2829
self.sim1.input_data.design_scale = dmat
30+
self.sim1.input_data._design_loc_names = new_coef_names
31+
self.sim1.input_data._design_scale_names = new_coef_names
2932
self.sim1.input_data.constraints_loc = constraints
3033
self.sim1.input_data.constraints_scale = constraints
3134

batchglm/unit_test/test_acc_glm_all.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,8 @@ def __init__(
5050
data=scipy.sparse.csr_matrix(simulator.input_data.x),
5151
design_loc=simulator.input_data.design_loc,
5252
design_scale=simulator.input_data.design_scale,
53+
design_loc_names=simulator.input_data.design_loc_names,
54+
design_scale_names=simulator.input_data.design_scale_names,
5355
constraints_loc=simulator.input_data.constraints_loc,
5456
constraints_scale=simulator.input_data.constraints_scale,
5557
size_factors=simulator.input_data.size_factors
@@ -59,6 +61,8 @@ def __init__(
5961
data=simulator.input_data.x,
6062
design_loc=simulator.input_data.design_loc,
6163
design_scale=simulator.input_data.design_scale,
64+
design_loc_names=simulator.input_data.design_loc_names,
65+
design_scale_names=simulator.input_data.design_scale_names,
6266
constraints_loc=simulator.input_data.constraints_loc,
6367
constraints_scale=simulator.input_data.constraints_scale,
6468
size_factors=simulator.input_data.size_factors

0 commit comments

Comments
 (0)