1 files changed, 54 insertions, 0 deletions
diff --git a/py/tests/helpers.py b/py/tests/helpers.py
new file mode 100644
index 0000000..fd67d41
--- /dev/null
+++ b/py/tests/helpers.py
@@ -0,0 +1,54 @@
+import pykms
+
+def add_props(areq, ob, map):
+    for key, value in map.items():
+        areq.add(ob, key, value)
+
+def props(o):
+    o.refresh_props()
+    map = o.prop_map
+    for propid,propval in map.items():
+        prop = o.card.get_prop(propid)
+        print("%-15s %d (%#x)" % (prop.name, propval, propval))
+
+def set_prop(ob, prop, value):
+    if ob.card.has_atomic:
+        areq = pykms.AtomicReq(ob.card)
+        areq.add(ob, prop, value)
+        if areq.commit_sync() != 0:
+            print("commit failed")
+    else:
+        if ob.set_prop_value(prop, value) != 0:
+            print("setting property failed")
+
+def set_props(ob, map):
+    if ob.card.has_atomic:
+        areq = pykms.AtomicReq(ob.card)
+
+        for key, value in map.items():
+            areq.add(ob, key, value)
+
+        if areq.commit_sync() != 0:
+            print("commit failed")
+    else:
+        for propid,propval in map.items():
+            if ob.set_prop_value(propid, propval) != 0:
+                print("setting property failed")
+
+red = pykms.RGB(255, 0, 0)
+green = pykms.RGB(0, 255, 0)
+blue = pykms.RGB(0, 0, 255)
+yellow = pykms.RGB(255, 255, 0)
+purple = pykms.RGB(255, 0, 255)
+white = pykms.RGB(255, 255, 255)
+cyan = pykms.RGB(0, 255, 255)
+
+def disable_planes(card):
+    areq = pykms.AtomicReq(card)
+
+    for p in card.planes:
+        areq.add(p, "FB_ID", 0)
+        areq.add(p, "CRTC_ID", 0)
+
+    if areq.commit_sync() != 0:
+        print("disabling planes failed")