def it_raises_on_malformed_arrays(): with zest.raises(AssertionError): im = imstack[:, 0, 0] imops.dump("name", im)
def it_converts_2d_arrays_to_3d(): im = imstack[:, :, 0] assert im.ndim == 2 imops.dump("name", im)
def it_dumps_to_local_path_if_root_not_set(): imops.dump_set_root_path(None) imops.dump("name", imstack) assert m_imstack_write.called_once_with(None, "name.npy", imstack)
def it_converts_lists_to_array_stacks(): list_stack = [imstack[:, :, 0], imstack[:, :, 1]] imops.dump("name", list_stack)
def it_dumps_to_the_root_path_if_set(): imops.dump_set_root_path("/dir") imops.dump("name", imstack) assert m_imstack_write.called_once_with("/dir", "name.npy", imstack)