def pass_python_data_to_prepared_script(
self, sds, var_name: str, prepared_script: JavaObject) -> None:
assert self.is_python_local_data, 'Can only pass data to prepared script if it is python local!'
if self._is_numpy():
prepared_script.setMatrix(var_name,
numpy_to_matrix_block(
sds, self._np_array),
True) # True for reuse
def convert_back_and_forth(self, array):
matrix_block = numpy_to_matrix_block(self.jvm, array)
# use the ability to call functions on matrix_block.
returned = matrix_block_to_numpy(self.jvm, matrix_block)
self.assertTrue(np.allclose(array, returned))
