@@ -98,15 +98,15 @@ def test_expansion_derivatives_and_powers():
98
98
{
99
99
"a" : [- 1 , - 2 , - 3 , - 4 , - 5 ],
100
100
"a_power2" : [1 , 4 , 9 , 16 , 25 ],
101
- "a_derivative1" : [np .NaN , - 1 , - 1 , - 1 , - 1 ],
102
- "a_derivative1_power2" : [np .NaN , 1 , 1 , 1 , 1 ],
103
- "b_derivative1" : [np .NaN , 0 , 0 , 0 , 0 ],
104
- "b_derivative1_power2" : [np .NaN , 0 , 0 , 0 , 0 ],
105
- "c_power2_derivative1" : [np .NaN , 1 , - 1 , 1 , - 1 ],
106
- "c_power2_derivative2" : [np .NaN , np .NaN , - 2 , 2 , - 2 ],
101
+ "a_derivative1" : [np .nan , - 1 , - 1 , - 1 , - 1 ],
102
+ "a_derivative1_power2" : [np .nan , 1 , 1 , 1 , 1 ],
103
+ "b_derivative1" : [np .nan , 0 , 0 , 0 , 0 ],
104
+ "b_derivative1_power2" : [np .nan , 0 , 0 , 0 , 0 ],
105
+ "c_power2_derivative1" : [np .nan , 1 , - 1 , 1 , - 1 ],
106
+ "c_power2_derivative2" : [np .nan , np .nan , - 2 , 2 , - 2 ],
107
107
"d" : [9 , 7 , 5 , 3 , 1 ],
108
108
"e" : [0 , 0 , 0 , 0 , 0 ],
109
- "f" : [np .NaN , 6 , 4 , 2 , 0 ],
109
+ "f" : [np .nan , 6 , 4 , 2 , 0 ],
110
110
}
111
111
)
112
112
exp_data = _expand_test (model_formula )
@@ -122,10 +122,10 @@ def test_expansion_na_robustness():
122
122
model_formula = "(dd1(f))^^2"
123
123
expected_data = pd .DataFrame (
124
124
{
125
- "f" : [np .NaN , 6 , 4 , 2 , 0 ],
126
- "f_power2" : [np .NaN , 36 , 16 , 4 , 0 ],
127
- "f_derivative1" : [np .NaN , np .NaN , - 2 , - 2 , - 2 ],
128
- "f_derivative1_power2" : [np .NaN , np .NaN , 4 , 4 , 4 ],
125
+ "f" : [np .nan , 6 , 4 , 2 , 0 ],
126
+ "f_power2" : [np .nan , 36 , 16 , 4 , 0 ],
127
+ "f_derivative1" : [np .nan , np .nan , - 2 , - 2 , - 2 ],
128
+ "f_derivative1_power2" : [np .nan , np .nan , 4 , 4 , 4 ],
129
129
}
130
130
)
131
131
exp_data = _expand_test (model_formula )
0 commit comments