def test_can_normalize_to_unit_area(self): kernel = sigproc.Kernel(1.0, normalize=False) kernel._evaluate = lambda x, _: sp.ones(len(x)) sampling_rate = 2.0 num_bins = 20 mock_discretization = sp.ones(num_bins) / num_bins * sampling_rate actual = sigproc.discretize_kernel( kernel, sampling_rate, num_bins=num_bins, ensure_unit_area=True) assert_array_equal(actual, mock_discretization)
def test_discretizes_requested_number_of_bins(self): kernel = sigproc.Kernel(1.0, normalize=False) kernel._evaluate = lambda x, _: sp.ones(len(x)) sampling_rate = 1.0 num_bins = 23 mock_discretization = sp.ones(num_bins) actual = sigproc.discretize_kernel( kernel, sampling_rate, num_bins=num_bins) assert_array_equal(actual, mock_discretization)
def test_discretizes_requested_number_of_bins(self): kernel = sigproc.Kernel(1.0, normalize=False) kernel._evaluate = lambda x, _: sp.ones(len(x)) sampling_rate = 1.0 num_bins = 23 mock_discretization = sp.ones(num_bins) actual = sigproc.discretize_kernel(kernel, sampling_rate, num_bins=num_bins) assert_array_equal(actual, mock_discretization)
def test_can_normalize_to_unit_area(self): kernel = sigproc.Kernel(1.0, normalize=False) kernel._evaluate = lambda x, _: sp.ones(len(x)) sampling_rate = 2.0 num_bins = 20 mock_discretization = sp.ones(num_bins) / num_bins * sampling_rate actual = sigproc.discretize_kernel(kernel, sampling_rate, num_bins=num_bins, ensure_unit_area=True) assert_array_equal(actual, mock_discretization)
def test_discretizes_requested_area_with_units(self): kernel = sigproc.Kernel(100.0 * pq.ms, normalize=False) kernel.boundary_enclosing_at_least = MagicMock( name='boundary_enclosing_at_least') kernel.boundary_enclosing_at_least.return_value = 100.0 * pq.ms kernel._evaluate = lambda x, _: sp.ones(len(x)) sampling_rate = 10.0 * pq.Hz mock_discretization = sp.ones(3) kernel_area_fraction = 0.5 actual = sigproc.discretize_kernel(kernel, sampling_rate, kernel_area_fraction) kernel.boundary_enclosing_at_least.assert_called_with( kernel_area_fraction) assert_array_equal(actual, mock_discretization)
def test_discretizes_requested_area_with_units(self): kernel = sigproc.Kernel(100.0 * pq.ms, normalize=False) kernel.boundary_enclosing_at_least = MagicMock( name='boundary_enclosing_at_least') kernel.boundary_enclosing_at_least.return_value = 100.0 * pq.ms kernel._evaluate = lambda x, _: sp.ones(len(x)) sampling_rate = 10.0 * pq.Hz mock_discretization = sp.ones(3) kernel_area_fraction = 0.5 actual = sigproc.discretize_kernel( kernel, sampling_rate, kernel_area_fraction) kernel.boundary_enclosing_at_least.assert_called_with( kernel_area_fraction) assert_array_equal(actual, mock_discretization)