def test_castToFloat(self):
arys, shape, size = _generateTestArrays(2, "uint8")
imageData = ImagesLoader(self.sc).fromArrays(arys)
catData = imageData.astype("smallfloat")
assert_equals("float16", str(catData.dtype))
assert_equals("float16", str(catData.first()[1].dtype))
def test_castToFloat(self):
arys, shape, size = _generateTestArrays(2, 'uint8')
imageData = ImagesLoader(self.sc).fromArrays(arys)
catData = imageData.astype("smallfloat")
assert_equals('float16', str(catData.dtype))
assert_equals('float16', str(catData.first()[1].dtype))
def test_castToFloat(self):
arys, shape, size = _generate_test_arrays(2, 'uint8')
imagedata = ImagesLoader(self.sc).fromArrays(arys)
castdata = imagedata.astype("smallfloat")
assert_equals('float16', str(castdata.dtype))
assert_equals('float16', str(castdata.first()[1].dtype))
