Revamp snowflake example notebook

Author: adamnsch

docs/source/integration.rst

@@ -19,7 +19,7 @@ Pandas DataFrames
 -----------------
 
 The ``neo4j-viz`` library provides a convenience method for importing data from Pandas DataFrames.
-These DataFrames can be created from many sources, such as CSV files or :doc:`Snowflake tables<./tutorials/snowpark-example>`.
+These DataFrames can be created from many sources, such as CSV files or :doc:`Snowflake tables<./tutorials/snowflake-example>`.
 It requires and additional dependency to be installed, which you can do by running:
 
 .. code-block:: bash
@@ -83,7 +83,7 @@ and :doc:`Relationships <./api-reference/relationship>`.
     VG = from_dfs(nodes, relationships)
 
 For another example of the ``from_dfs`` importer in action, see the
-:doc:`Visualizing Snowflake Tables tutorial <./tutorials/snowpark-example>`.
+:doc:`Visualizing Snowflake Tables tutorial <./tutorials/snowflake-example>`.
 
 
 Neo4j Graph Data Science (GDS) library

docs/source/tutorials/snowflake-example.nblink

@@ -0,0 +1,3 @@
+{
+  "path": "../../../examples/snowflake-example.ipynb"
+}

docs/source/tutorials/snowpark-example.nblink

@@ -37,6 +37,7 @@
 )
 
 from neo4j_viz import VisualizationGraph
+from neo4j_viz.colors import ColorSpace
 from neo4j_viz.pandas import from_dfs
 
 
@@ -336,4 +337,8 @@ def from_snowflake(
         for i, rel_df in enumerate(rel_dfs):
             rel_df["caption"] = rel_table_names[i].split(".")[-1]
 
-    return from_dfs(node_dfs, rel_dfs, node_radius_min_max)
+    VG = from_dfs(node_dfs, rel_dfs, node_radius_min_max)
+
+    VG.color_nodes(field="caption", color_space=ColorSpace.DISCRETE)
+
+    return VG

examples/snowflake-example.ipynb

@@ -62,10 +62,12 @@ def test_from_snowflake(session_with_minimal_graph: Session) -> None:
 
     assert VG.nodes[0].id == 0
     assert VG.nodes[0].caption == "NODES"
+    assert VG.nodes[0].color is not None
     assert VG.nodes[0].properties == {"SNOWFLAKEID": 6}
 
     assert VG.nodes[1].id == 1
     assert VG.nodes[1].caption == "NODES"
+    assert VG.nodes[0].color is not None
     assert VG.nodes[1].properties == {"SNOWFLAKEID": 7}
 
     assert len(VG.relationships) == 1