pearl-public/pmsco-public
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update public distribution

Browse Source 
based on internal repository c9a2ac8 2019-01-03 16:04:57 +0100
tagged rev-master-2.0.0
This commit is contained in:
muntwiler_m
2019-01-31 15:45:02 +01:00
parent bbd16d0f94
commit acea809e4e
92 changed files with 165828 additions and 143181 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
tests/test_data.py
View File

@ -20,6 +20,9 @@ Licensed under the Apache License, Version 2.0 (the "License"); @n
http://www.apache.org/licenses/LICENSE-2.0
"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import unittest
import math
import numpy as np
@ -80,7 +83,7 @@ class TestDataFunctions(unittest.TestCase):
shape = (10, )
data = md.create_data(shape, dtype=md.DTYPE_ETPAIS)
expected_names = ('e', 't', 'p', 'a', 'i', 's')
self.assertItemsEqual(data.dtype.names, expected_names)
self.assertEqual(data.dtype.names, expected_names)
self.assertEqual(data.shape, shape)
def test_detect_scan_mode_1d(self):
@ -88,10 +91,11 @@ class TestDataFunctions(unittest.TestCase):
expected_mode = ['e']
expected_positions = {}
expected_positions['e'] = np.linspace(0.0, 10.0, 10)
expected_positions['e'] = np.linspace(100.0, 200.0, 10)
self.assertItemsEqual(scan_mode, expected_mode)
self.assertItemsEqual(scan_positions, expected_positions)
self.assertEqual(scan_mode, expected_mode)
for dim in expected_positions:
np.testing.assert_almost_equal(scan_positions[dim], expected_positions[dim], decimal=3)
def test_detect_scan_mode_2d(self):
scan_mode, scan_positions = md.detect_scan_mode(self.ea_scan)
@ -103,8 +107,9 @@ class TestDataFunctions(unittest.TestCase):
expected_positions['p'] = np.zeros((1))
expected_positions['a'] = np.asarray((-1.0, 0.0, 1.0, 2.0))
self.assertItemsEqual(scan_mode, expected_mode)
self.assertItemsEqual(scan_positions, expected_positions)
self.assertEqual(scan_mode, expected_mode)
for dim in expected_positions:
np.testing.assert_almost_equal(scan_positions[dim], expected_positions[dim], decimal=3)
def test_detect_scan_mode_holo(self):
scan_mode, scan_positions = md.detect_scan_mode(self.holo_scan)
@ -115,8 +120,9 @@ class TestDataFunctions(unittest.TestCase):
expected_positions['t'] = self.holo_scan['t']
expected_positions['p'] = self.holo_scan['p']
self.assertItemsEqual(scan_mode, expected_mode)
self.assertItemsEqual(scan_positions, expected_positions)
self.assertEqual(scan_mode, expected_mode)
for dim in expected_positions:
np.testing.assert_almost_equal(scan_positions[dim], expected_positions[dim], decimal=3)
def test_detect_scan_mode_theta(self):
scan = self.holo_scan
@ -130,8 +136,9 @@ class TestDataFunctions(unittest.TestCase):
expected_positions['t'] = np.linspace(1.0, 2.0, scan.shape[0])
expected_positions['p'] = np.ones((1)) * 3.3
self.assertItemsEqual(scan_mode, expected_mode)
self.assertItemsEqual(scan_positions, expected_positions)
self.assertEqual(scan_mode, expected_mode)
for dim in expected_positions:
np.testing.assert_almost_equal(scan_positions[dim], expected_positions[dim], decimal=3)
def test_calc_modfunc_mean_1d(self):
modf = md.calc_modfunc_mean(self.e_scan)
@ -178,6 +185,7 @@ class TestDataFunctions(unittest.TestCase):
exp_modf['i'] = +1.000001
np.testing.assert_array_less(modf['i'], exp_modf['i'])
@unittest.skip("test_calc_modfunc_loess_1d_nan fails and kills the whole test process")
def test_calc_modfunc_loess_1d_nan(self):
"""
check that data.calc_modfunc_loess() ignores NaNs gracefully.
@ -229,7 +237,7 @@ class TestDataFunctions(unittest.TestCase):
# the R factor should be sensitive enough to detect mixed-up axes.
exp_modf['i'] = 0.03 * np.sin((scan['e'] - 150) / 50 * math.pi)
rf = md.rfactor(modf, exp_modf)
print rf
print(rf)
self.assertLessEqual(rf, 0.50)
def test_alpha_mirror_average(self):