Merge pull request #68 from CitrineInformatics/add-image

Add image for generating a filelink

Author: kroenlein

taurus/demo/cake.py

@@ -47,6 +47,14 @@
 DEMO_SCOPE = 'citrine-demo'
 
 
+def import_toothpick_picture():
+    """Return the stream of the toothpick picture."""
+    import pkg_resources
+    resource = pkg_resources.resource_stream("taurus.demo", "toothpick.jpg")
+
+    return resource
+
+
 def make_cake_templates():
     """Define all templates independently, as in the wild this will be an independent operation."""
     tmpl = dict()

taurus/demo/tests/test_cake.py

@@ -9,7 +9,7 @@
 from taurus.entity.object.ingredient_run import IngredientRun
 
 from taurus.client.json_encoder import dumps, loads
-from taurus.demo.cake import make_cake
+from taurus.demo.cake import make_cake, import_toothpick_picture
 from taurus.util import recursive_foreach
 from taurus.entity.util import complete_material_history
 
@@ -112,3 +112,8 @@ def check_ids(obj):
                     assert obj.spec.material == obj.material.spec
             if obj.material:
                 queue.append(obj.material)
+
+
+def test_import():
+    """Make sure picture import runs."""
+    import_toothpick_picture()