def test_ctf_manager_load(): transfer_func = TransferFunction() with temp_directory() as root_dir: manager = CtfManager.from_directory(root_dir) manager.add("test", transfer_func) del manager manager = CtfManager.from_directory(root_dir) manager.get("test")
def test_ctf_manager_load(): transfer_func = TransferFunction() with temp_directory() as root_dir: manager = CtfManager.from_directory(root_dir) manager.add('test', transfer_func) del manager manager = CtfManager.from_directory(root_dir) manager.get('test')
def _test_ctf_manager_names(names_to_test): transfer_func = TransferFunction() with temp_directory() as root_dir: manager = CtfManager.from_directory(root_dir) for name in names_to_test: manager.add(name, transfer_func) # Reload and check the names manager = CtfManager.from_directory(root_dir) for name in manager.names: assert name in names_to_test
def test_ctf_manager_add(): transfer_func = TransferFunction() with temp_directory() as root_dir: manager = CtfManager.from_directory(root_dir) for i in range(5): manager.add(str(i), transfer_func) assert len(listdir(root_dir)) == 5
def test_ctf_manager_get(): transfer_func = TransferFunction() with temp_directory() as root_dir: manager = CtfManager.from_directory(root_dir) manager.add("test", transfer_func) ret_func = manager.get("test") assert_allclose(ret_func.color.values(), transfer_func.color.values()) assert_allclose(ret_func.opacity.values(), transfer_func.opacity.values())
def test_ctf_manager_get(): transfer_func = TransferFunction() with temp_directory() as root_dir: manager = CtfManager.from_directory(root_dir) manager.add('test', transfer_func) ret_func = manager.get('test') assert_allclose(ret_func.color.values(), transfer_func.color.values()) assert_allclose(ret_func.opacity.values(), transfer_func.opacity.values())
To use: right-click in the window to bring up a context menu. Once you've added a color or opacity, you can drag them around by just clicking on them. The colors at the end points are editable, but cannot be removed. """ from os.path import join from traits.etsconfig.api import ETSConfig ETSConfig.toolkit = 'qt4' from enaml.qt.qt_application import QtApplication from ensemble.ctf.api import CtfEditor, CtfManager, get_color import traits_enaml if __name__ == "__main__": with traits_enaml.imports(): from ctf_demo_window import CtfDemoWindow app = QtApplication() ctf_editor = CtfEditor(prompt_color_selection=get_color) ctf_manager = CtfManager.from_directory( join(ETSConfig.application_data, 'CTF_demo') ) win = CtfDemoWindow(ctf_editor=ctf_editor, ctf_manager=ctf_manager) win.show() app.start()