add unit tests of NumpyTupleDatasetFeatureIndexer

chainer · Jun 26, 2018 · 7a70195 · 7a70195
1 parent 94cf8f3
commit 7a70195
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/dataset_tests/test_numpy_tuple_feature_indexer.py b/tests/dataset_tests/test_numpy_tuple_feature_indexer.py
@@ -33,6 +33,14 @@ def test_extract_feature_by_slice(self, indexer, data, slice_index, j):
             indexer.extract_feature_by_slice(slice_index, j),
             data[j][slice_index])
 
+    @pytest.mark.parametrize('ndarray_index', [numpy.asarray([0, 1]),
+                                               numpy.asarray([1])])
+    @pytest.mark.parametrize('j', [0, 1])
+    def test_extract_feature_by_ndarray(self, indexer, data, ndarray_index, j):
+        numpy.testing.assert_array_equal(
+            indexer.extract_feature_by_slice(ndarray_index, j),
+            data[j][ndarray_index])
+
 
 if __name__ == '__main__':
     pytest.main([__file__, '-v'])