python tests, includes fixes

rapidsai · Jul 9, 2021 · d320e1f · d320e1f
1 parent 87072d7
commit d320e1f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/cpp/tests/replace/replace_nulls_tests.cpp b/cpp/tests/replace/replace_nulls_tests.cpp
@@ -33,7 +33,8 @@
 #include <cudf_test/cudf_gtest.hpp>
 #include <cudf_test/iterator_utilities.hpp>
 #include <cudf_test/type_lists.hpp>
-#include "thrust/iterator/counting_iterator.h"
+
+#include <thrust/iterator/counting_iterator.h>
 
 using namespace cudf::test::iterators;
 

diff --git a/python/cudf/cudf/tests/test_replace.py b/python/cudf/cudf/tests/test_replace.py
@@ -303,6 +303,8 @@ def test_series_fillna_numerical(psr, data_dtype, fill_value, inplace):
         [1, None, None, 2, 3, 4],
         [None, None, 1, 2, None, 3, 4],
         [1, 2, None, 3, 4, None, None],
+        [0] + [None] * 14,
+        [None] * 14 + [0],
     ],
 )
 @pytest.mark.parametrize("container", [pd.Series, pd.DataFrame])