Adding registration doctests

.gitlab-ci.yml

@@ -2,3 +2,5 @@ unittests:
     script: python -m unittest discover tests -v
 core doctests:
     script: python -m doctest -v AmpScan/core.py
+registration doctests:
+    script: python -m doctest -v AmpScan/registration.py
\ No newline at end of file

AmpScan/registration.py

@@ -6,7 +6,7 @@ Copyright: Joshua Steer 2018, Joshua.Steer@soton.ac.uk
 import numpy as np
 import copy
 from scipy import spatial
-from .core import AmpObject
+from AmpScan.core import AmpObject
 import matplotlib.pyplot as plt
 
 class registration(object):
@@ -36,9 +36,13 @@ class registration(object):
     
     Examples
     --------
+    >>> import os
+    >>> import AmpScan
+    >>> basefh = os.getcwd()+"\\tests\\stl_file.stl"
+    >>> targfh = os.getcwd()+"\\tests\\stl_file_2.stl"
     >>> baseline = AmpScan.AmpObject(basefh)
     >>> target = AmpScan.AmpObject(targfh)
-    >>> reg = AmpScan.registration(steps=10, neigh=10, smooth=1).reg
+    >>> reg = AmpScan.registration(baseline, target, steps=10, neigh=10, smooth=1).reg
 		
     """ 
     def __init__(self, baseline, target, method='point2plane', *args, **kwargs):

tests/doctests.py

-import AmpScan.core as core
-import AmpScan.trim as trim
-
-if __name__ == '__main__':
-    import doctest
-    doctest.testmod(core, verbose=True)
-    doctest.testmod(trim, verbose=True)
-