def test_init(self): self.mf = process_traces.MeanFilter(use_threshold=True, threshold=sentinel.threshold) self.assertEqual(self.mf.threshold, sentinel.threshold) self.assertEqual(self.mf.filter, self.mf.mean_filter_with_threshold) self.mf = process_traces.MeanFilter(use_threshold=False) self.assertRaises(AttributeError, lambda: self.mf.threshold) self.assertEqual(self.mf.filter, self.mf.mean_filter_without_threshold)
def setUp(self): self.trace = [[200] * 400 + [500] + [400] * 20 + [200] * 600] self.traces = self.trace * 2 self.mf = process_traces.MeanFilter()