From 3b89309b8b279c4e4edb31edce955ff14edc05e7 Mon Sep 17 00:00:00 2001
From: daniel <bauer@cbs.tu-darmstadt.de>
Date: Tue, 17 Nov 2020 08:41:11 +0100
Subject: [PATCH] test filter coords

---
 src/frame.rs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/frame.rs b/src/frame.rs
index f8b6c50..6b9f0e9 100644
--- a/src/frame.rs
+++ b/src/frame.rs
@@ -97,4 +97,19 @@ mod tests {
         let frame = Frame::with_len(10);
         assert_eq!(frame.len(), 10);
     }
+
+    #[test]
+    fn test_filter_coords() {
+        let mut frame = Frame {
+            step: 0,
+            time: 0.0,
+            box_vector: [[0.0; 3]; 3],
+            coords: vec![[0.0; 3], [1.0; 3], [2.0; 3]]
+        };
+
+        frame.filter_coords(&[1]);
+        for i in 0..3 {
+            assert_approx_eq!(frame.coords[0][i], 1.0);
+        }
+    }
 }