From 73b415bd8172b7ee9dc06b0ff68b9a407a96a72f Mon Sep 17 00:00:00 2001
From: Paul McCarthy <pauldmccarthy@gmail.com>
Date: Thu, 30 Nov 2017 13:20:10 +1100
Subject: [PATCH] unit tests for rmsdev function
---
tests/test_transform.py | 31 +++++++++++++++++++++++++++++++
1 file changed, 31 insertions(+)
diff --git a/tests/test_transform.py b/tests/test_transform.py
index 0ed50c64e..a6688b85f 100644
--- a/tests/test_transform.py
+++ b/tests/test_transform.py
@@ -537,3 +537,34 @@ def test_sformToFlirtMatrix():
assert np.all(np.isclose(result1, expected))
assert np.all(np.isclose(result2, expected))
+
+
+def test_rmsdev():
+
+ t1 = np.eye(4)
+ t2 = transform.scaleOffsetXform([1, 1, 1], [2, 0, 0])
+
+ assert np.isclose(transform.rmsdev(t1, t2), 2)
+ assert np.isclose(transform.rmsdev(t1, t2, R=2), 2)
+ assert np.isclose(transform.rmsdev(t1, t2, R=2, xc=(1, 1, 1)), 2)
+
+ t1 = np.eye(3)
+ lastdist = 0
+
+ for i in range(1, 11):
+ rot = np.pi * i / 10.0
+ t2 = transform.axisAnglesToRotMat(rot, 0, 0)
+ result = transform.rmsdev(t1, t2)
+
+ assert result > lastdist
+
+ lastdist = result
+
+ for i in range(11, 20):
+ rot = np.pi * i / 10.0
+ t2 = transform.axisAnglesToRotMat(rot, 0, 0)
+ result = transform.rmsdev(t1, t2)
+
+ assert result < lastdist
+
+ lastdist = result
--
GitLab