def test_init_dict_with_empty_directory(self):
with tempfile.TemporaryDirectory() as directory:
model = demo.DemoPath(name=directory, )
# traits api is still imported.
init_dic = model.init_dic
self.assertIsInstance(init_dic, dict)
self.assertIn("HasTraits", init_dic)
def test_init_dict_with_init_py(self):
with tempfile.TemporaryDirectory() as directory:
init_py = os.path.join(directory, "__init__.py")
with open(init_py, "w", encoding="utf-8") as f:
f.write("a = 1\n")
model = demo.DemoPath(name=directory, )
self.assertIn("a", model.init_dic)
self.assertEqual(model.init_dic["a"], 1)
# traits api is still imported.
self.assertIn("HasTraits", model.init_dic)
def test_description_with_empty_directory(self):
# If the directory is empty, the content of the description should
# be empty.
with tempfile.TemporaryDirectory() as directory:
model = demo.DemoPath(name=directory, )
tree = ET.fromstring(model.description)
body_node = next(tree.iter(get_html_tag("body")))
div_node, = list(body_node)
self.assertEqual(list(div_node), [])
def test_description_use_css(self):
with tempfile.TemporaryDirectory() as directory:
model = demo.DemoPath(
name=directory,
css_filename="default.css",
)
tree = ET.fromstring(model.description)
link_node = next(tree.iter(get_html_tag("link")))
self.assertEqual(link_node.attrib["href"], "default.css")
def test_use_index_rst(self):
with tempfile.TemporaryDirectory() as directory:
index_rst = os.path.join(directory, "index.rst")
with open(index_rst, "w", encoding="utf-8") as f:
f.write(".. image:: any_image.jpg\n")
model = demo.DemoPath(name=directory, )
tree = ET.fromstring(model.description)
img_node = next(tree.iter(get_html_tag("img")))
self.assertEqual(img_node.attrib["src"], "any_image.jpg")
def test_init_dict_used_by_children(self):
# Test the __init__.py in a directory (if exists) is visible
# by the example scripts in that folder.
# Not sure if this is really needed, but this is an existing feature.
with tempfile.TemporaryDirectory() as directory:
init_content = "CONSTANT = 'HELLO'"
example_content = textwrap.dedent("""
from . import CONSTANT
from . import HasTraits
""")
subdir = os.path.join(directory, "Examples")
os.makedirs(subdir)
init_py = os.path.join(subdir, "__init__.py")
with open(init_py, "w", encoding="utf-8") as f:
f.write(init_content)
example_py = os.path.join(subdir, "example.py")
with open(example_py, "w", encoding="utf-8") as f:
f.write(example_content)
model = demo.DemoPath(
name=directory,
use_files=False,
)
# sanity check:
# This is one subdirectory
children = model.get_children()
self.assertEqual(len(children), 1)
# In that subdirectory, there is one Python file that is not
# __init__.py
subdir_node, = children
file_nodes = subdir_node.get_children()
self.assertEqual(len(file_nodes), 1)
example, = file_nodes
self.assertEqual(example.name, "example.py")
# This is the test objective: The __init__.py and traits api are
# loaded and accessible by the example script.
# Try running the code
example.run_code()
self.assertIn("CONSTANT", example.locals)
self.assertIn("HasTraits", example.locals)