def test_dim2array(self):
class MockIMDDimension:
def __init__(self):
self.getMinimum = Mock(return_value=1)
self.getMaximum = Mock(return_value=10)
self.getNBins = Mock(return_value=3)
mock_dimension = MockIMDDimension()
result = funcs._dim2array(mock_dimension)
self.assertTrue(np.array_equal([1., 4., 7., 10.], result))
def test_dim2array(self):
class MockIMDDimension:
def __init__(self):
self.getMinimum = Mock(return_value=1)
self.getMaximum = Mock(return_value=10)
self.getNBins = Mock(return_value=3)
mock_dimension = MockIMDDimension()
result = funcs._dim2array(mock_dimension)
self.assertTrue(np.array_equal([1., 4., 7., 10.], result))