def test_calcEstimates_with_nonzero_analytical_solutions(self):
wrMock = MagicMock(return_value=1)
with patch('miscFuncs.calcStatsWrapper', wrMock):
an = AnalysisNodeData('node', ['K1', 'K3'], 'areas')
an.angles = [1, 2]
an.anSol = {'K1': [1, 2], 'K3': [1, 2]}
an.results = {'K1': [1, 2], 'K3': [1, 2]}
an.calcEstimates()
expected = {
e: {
'K1': 1,
'K3': 1
}
for e in
['areaDiff', 'dotProd', 'avgNormError', 'maxNormError', 'rmsd']
}
self.assertEqual(expected, an.estimates)
def test_fixAngleValues(self):
an = AnalysisNodeData('node', ['K1'], 'areas')
an.angles = np.array([1, 4, 90, 120, 90, 340, 380, 720])
an.fixAngleValues()
exp = [1, 4, 90, 120, 90, 340, 20, 0]
self.assertEqual(exp, list(an.angles))