tidy

Author: ljeub-pometry

python/python/raphtory/__init__.pyi

@@ -444,6 +444,15 @@ class Edge(object):
 
         """
 
+    def history_counts(self) -> int:
+        """
+        Returns the number of times an edge is added or change to an edge is made.
+
+        Returns:
+           int: The number of times an edge is added or change to an edge is made.
+
+        """
+
     def history_date_time(self):
         """
         Returns a list of timestamps of when an edge is added or change to an edge is made.
@@ -895,6 +904,7 @@ class Edges(object):
 
         """
 
+    def history_counts(self): ...
     def history_date_time(self):
         """
         Returns all timestamps of edges, when an edge is added or change to an edge is made.

python/tests/graphql/update_graph/test_batch_updates.py

@@ -13,6 +13,7 @@
     RemoteEdgeAddition,
 )
 
+
 def make_props():
     current_datetime = datetime.now(timezone.utc)
     naive_datetime = datetime.now()
@@ -116,12 +117,15 @@ def test_add_nodes():
         ben = g.node("ben")
         hamza = g.node("hamza")
         lucas = g.node("lucas")
-        check_arr(ben.properties.temporal.get("prop_float").values(), [
-            2.0,
-            3.0,
-            2.0,
-            3.0,
-        ])
+        check_arr(
+            ben.properties.temporal.get("prop_float").values(),
+            [
+                2.0,
+                3.0,
+                2.0,
+                3.0,
+            ],
+        )
         check_arr(ben.history(), [1, 2, 3, 4, 5, 6])
         assert ben.node_type == "person"
         check_arr(hamza.history(), [1, 2])
@@ -176,14 +180,17 @@ def test_add_edges():
         ben_hammza = g.edge("ben", "hamza")
         hamza_lucas = g.edge("hamza", "lucas")
         lucas_hamza = g.edge("lucas", "hamza")
-        check_arr( ben_hammza.properties.temporal.get("prop_float").values() , [
-            2.0,
-            3.0,
-            2.0,
-            3.0,
-        ])
-        check_arr( ben_hammza.history() , [1, 2, 3, 4, 5, 6])
+        check_arr(
+            ben_hammza.properties.temporal.get("prop_float").values(),
+            [
+                2.0,
+                3.0,
+                2.0,
+                3.0,
+            ],
+        )
+        check_arr(ben_hammza.history(), [1, 2, 3, 4, 5, 6])
         assert ben_hammza.layer_names == ["_default", "test"]
-        check_arr( hamza_lucas.history() , [1, 2])
+        check_arr(hamza_lucas.history(), [1, 2])
         assert hamza_lucas.layer_names == ["_default"]
         helper_test_props(lucas_hamza.layer("_default"), lucas_props)

python/tests/graphql/update_graph/test_edge_updates.py

@@ -5,6 +5,7 @@
 from raphtory.graphql import GraphServer, RaphtoryClient
 from numpy.testing import assert_equal as check_arr
 
+
 def make_props():
     current_datetime = datetime.now(timezone.utc)
     naive_datetime = datetime.now()
@@ -76,9 +77,9 @@ def test_add_updates():
         g = client.receive_graph("path/to/event_graph")
         e = g.edge("ben", "hamza")
         helper_test_props(e, props)
-        check_arr( e.properties.temporal.get("prop_float").history() , [2, 3])
-        check_arr( e.layer("test").properties.temporal.get("prop_float").history() , [2])
-        check_arr( e.history() , [1, 2, 3, 4, 5, 6])
+        check_arr(e.properties.temporal.get("prop_float").history(), [2, 3])
+        check_arr(e.layer("test").properties.temporal.get("prop_float").history(), [2])
+        check_arr(e.history(), [1, 2, 3, 4, 5, 6])
 
 
 def test_add_constant_properties():

python/tests/graphql/update_graph/test_node_updates.py

@@ -69,7 +69,7 @@ def test_add_updates():
         rg.node("ben").add_updates(4)
         g = client.receive_graph("path/to/event_graph")
         helper_test_props(g.node("ben"), props)
-        check_arr( g.node("ben").history() , [1, 2, 3, 4])
+        check_arr(g.node("ben").history(), [1, 2, 3, 4])
 
 
 def test_add_constant_properties():

python/tests/test_graphdb/test_latest_graph.py

@@ -157,7 +157,7 @@ def test_persistent_node_latest():
     wg = g.window(5, 12)
     assert wg.latest_time == 11
     assert wg.earliest_time == 5
-    check_arr( wg.node(1).latest().history() , [])
+    check_arr(wg.node(1).latest().history(), [])
 
-    check_arr( g.nodes.latest().id.collect() , [1, 2, 3])
-    check_arr( wg.nodes.latest().id.collect() , [1, 2])
+    check_arr(g.nodes.latest().id.collect(), [1, 2, 3])
+    check_arr(wg.nodes.latest().id.collect(), [1, 2])

raphtory/src/io/parquet_loaders.rs

@@ -12,6 +12,8 @@ use crate::{
     serialise::incremental::InternalCache,
 };
 use itertools::Itertools;
+#[cfg(feature = "storage")]
+use polars_arrow::array::StructArray;
 use polars_arrow::datatypes::{ArrowDataType as DataType, ArrowSchema, Field};
 use polars_parquet::{
     read,