Blame patsy-nopandas.patch
|
Sergio Pascual |
1f0362a |
diff -ur patsy-0.2.1/patsy/test_highlevel.py patsy-0.2.1.nopandas/patsy/test_highlevel.py
|
|
Sergio Pascual |
1f0362a |
--- patsy-0.2.1/patsy/test_highlevel.py 2014-01-09 22:45:10.795234808 +0100
|
|
Sergio Pascual |
1f0362a |
+++ patsy-0.2.1.nopandas/patsy/test_highlevel.py 2014-01-09 22:46:42.958974829 +0100
|
|
Sergio Pascual |
1f0362a |
@@ -645,7 +645,12 @@
|
|
Sergio Pascual |
1f0362a |
def test_dmatrix_NA_action():
|
|
Sergio Pascual |
1f0362a |
data = {"x": [1, 2, 3, np.nan], "y": [np.nan, 20, 30, 40]}
|
|
Sergio Pascual |
1f0362a |
|
|
Sergio Pascual |
1f0362a |
- for return_type in ["matrix", "dataframe"]:
|
|
Sergio Pascual |
1f0362a |
+ return_types = ["matrix"]
|
|
Sergio Pascual |
1f0362a |
+
|
|
Sergio Pascual |
1f0362a |
+ if have_pandas:
|
|
Sergio Pascual |
1f0362a |
+ return_types.append("dataframe")
|
|
Sergio Pascual |
1f0362a |
+
|
|
Sergio Pascual |
1f0362a |
+ for return_type in return_types:
|
|
Sergio Pascual |
1f0362a |
mat = dmatrix("x + y", data=data, return_type=return_type)
|
|
Sergio Pascual |
1f0362a |
assert np.array_equal(mat, [[1, 2, 20],
|
|
Sergio Pascual |
1f0362a |
[1, 3, 30]])
|