Exemple #1
0
 def test_linear_emsd(self):
     A = 1
     EARLY = 7 # only early lag times have good stats
     actual = mr.emsd(self.many_walks, 1, 1, max_lagtime=EARLY)
     a = np.arange(EARLY, dtype='float64')
     expected = Series(2*A*a, index=a).iloc[1:]
     expected.name = 'msd'
     expected.index.name = 'lag time [s]'
     # HACK: Float64Index imprecision ruins index equality.
     # Test them separately. If that works, make them exactly the same.
     assert_almost_equal(actual.index.values, expected.index.values)
     actual.index = expected.index
     assert_series_equal(np.round(actual), expected)
Exemple #2
0
 def test_zero_emsd(self):
     N = 10
     actual = mr.emsd(self.dead_still, 1, 1)
     expected = Series(np.zeros(N)).iloc[1:].astype('float64')
     assert_series_equal(actual, expected)