From ea02bb7ee3509c5f521641a2e613af366dcfdbc2 Mon Sep 17 00:00:00 2001
From: Jan Grewe <jan.grewe@g-node.org>
Date: Sat, 19 Dec 2020 18:13:17 +0100
Subject: [PATCH] add test script

---
 test.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)
 create mode 100644 test.py

diff --git a/test.py b/test.py
new file mode 100644
index 0000000..b224fa2
--- /dev/null
+++ b/test.py
@@ -0,0 +1,20 @@
+import matplotlib.pyplot as plt
+import numpy as np 
+from IPython import embed, embed_kernel
+from tracking_result import TrackingResult
+import image_marker as im
+
+
+def main(filename):
+    tr = TrackingResult(filename)
+    tr.plot(bodypart="snout")
+    tank_task = im.MarkerTask("tank limits", ["bottom left corner", "top left corner", "top right corner", "bottom right corner"], "Mark tank corners")
+    feeder_task = im.MarkerTask("Feeder positions", list(map(str, range(1, 2))), "Mark feeder positions")
+    tasks = [tank_task, feeder_task]
+    image_marker = im.ImageMarker(tasks)
+    marker_positions = image_marker.mark_movie(filename, 1)
+    print(marker_positions)
+
+if __name__ == "__main__":
+    filename = "2020.12.11_lepto48DLC_resnet50_boldnessDec11shuffle1_200000.h5"
+    main(filename)