fix: added union_no_overlap to query2 transforms

ActivityWatch · Jul 5, 2022 · 657aeec · 657aeec
1 parent c1bd040
commit 657aeec
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/aw_query/functions.py b/aw_query/functions.py
@@ -13,6 +13,7 @@
     filter_keyvals,
     filter_keyvals_regex,
     period_union,
+    union_no_overlap,
     categorize,
     tag,
     Rule,
@@ -266,6 +267,12 @@ def q2_concat(events1: list, events2: list) -> List[Event]:
     return concat(events1, events2)
 
 
+@q2_function(union_no_overlap)
+@q2_typecheck
+def q2_union_no_overlap(events1: list, events2: list) -> List[Event]:
+    return union_no_overlap(events1, events2)
+
+
 """
     Flood functions
 """

diff --git a/aw_transform/union_no_overlap.py b/aw_transform/union_no_overlap.py
@@ -30,6 +30,7 @@ def test_split_event():
     e1, e2 = _split_event(e, now + td1h)
     assert e1.timestamp == now
     assert e1.duration == td1h
+    assert e2
     assert e2.timestamp == now + td1h
     assert e2.duration == td1h