@@ -117,7 +117,7 @@ def test_SparseNdarray_check(mask_rate):
117117 contents = mock_SparseNdarray_contents (test_shape , mask_rate = mask_rate )
118118 y = delayedarray .SparseNdarray (test_shape , contents )
119119 assert y .shape == test_shape
120- assert y .dtype == numpy .float64
120+ assert y .dtype is numpy .dtype ( " float64" )
121121 assert repr (y ).find ("SparseNdarray" ) > 0
122122 assert delayedarray .is_sparse (y )
123123 assert delayedarray .is_masked (y ) == (mask_rate > 0 )
@@ -172,9 +172,13 @@ def shorten(con, depth):
172172
173173 empty = delayedarray .SparseNdarray (test_shape , None , dtype = numpy .dtype ("int32" ), index_dtype = numpy .dtype ("int32" ))
174174 assert empty .shape == test_shape
175- assert empty .dtype == numpy .int32
175+ assert empty .dtype is numpy .dtype ( " int32" )
176176 assert not empty .is_masked
177177
178+ empty = delayedarray .SparseNdarray (test_shape , None , dtype = numpy .float32 , index_dtype = numpy .int32 ) # generics converted to dtypes
179+ assert empty .dtype is numpy .dtype ("float32" )
180+ assert empty .index_dtype is numpy .dtype ("int32" )
181+
178182
179183#######################################################
180184#######################################################
0 commit comments