@@ -101,15 +101,15 @@ def test_expansion_derivatives_and_powers():
101101        {
102102            'a' : [- 1 , - 2 , - 3 , - 4 , - 5 ],
103103            'a_power2' : [1 , 4 , 9 , 16 , 25 ],
104-             'a_derivative1' : [np .NaN , - 1 , - 1 , - 1 , - 1 ],
105-             'a_derivative1_power2' : [np .NaN , 1 , 1 , 1 , 1 ],
106-             'b_derivative1' : [np .NaN , 0 , 0 , 0 , 0 ],
107-             'b_derivative1_power2' : [np .NaN , 0 , 0 , 0 , 0 ],
108-             'c_power2_derivative1' : [np .NaN , 1 , - 1 , 1 , - 1 ],
109-             'c_power2_derivative2' : [np .NaN , np .NaN , - 2 , 2 , - 2 ],
104+             'a_derivative1' : [np .nan , - 1 , - 1 , - 1 , - 1 ],
105+             'a_derivative1_power2' : [np .nan , 1 , 1 , 1 , 1 ],
106+             'b_derivative1' : [np .nan , 0 , 0 , 0 , 0 ],
107+             'b_derivative1_power2' : [np .nan , 0 , 0 , 0 , 0 ],
108+             'c_power2_derivative1' : [np .nan , 1 , - 1 , 1 , - 1 ],
109+             'c_power2_derivative2' : [np .nan , np .nan , - 2 , 2 , - 2 ],
110110            'd' : [9 , 7 , 5 , 3 , 1 ],
111111            'e' : [0 , 0 , 0 , 0 , 0 ],
112-             'f' : [np .NaN , 6 , 4 , 2 , 0 ],
112+             'f' : [np .nan , 6 , 4 , 2 , 0 ],
113113        }
114114    )
115115    exp_data  =  _expand_test (model_formula )
@@ -123,10 +123,10 @@ def test_expansion_na_robustness():
123123    model_formula  =  '(dd1(f))^^2' 
124124    expected_data  =  pd .DataFrame (
125125        {
126-             'f' : [np .NaN , 6 , 4 , 2 , 0 ],
127-             'f_power2' : [np .NaN , 36 , 16 , 4 , 0 ],
128-             'f_derivative1' : [np .NaN , np .NaN , - 2 , - 2 , - 2 ],
129-             'f_derivative1_power2' : [np .NaN , np .NaN , 4 , 4 , 4 ],
126+             'f' : [np .nan , 6 , 4 , 2 , 0 ],
127+             'f_power2' : [np .nan , 36 , 16 , 4 , 0 ],
128+             'f_derivative1' : [np .nan , np .nan , - 2 , - 2 , - 2 ],
129+             'f_derivative1_power2' : [np .nan , np .nan , 4 , 4 , 4 ],
130130        }
131131    )
132132    exp_data  =  _expand_test (model_formula )
0 commit comments