class TestDataArray(object):
"""Unit tests for DataArray"""
def setup_method(self, method):
"""Creates empty DataArray"""
self.data_array = DataArray((100, 100, 100))
def test_iter(self):
"""Unit test for __iter__"""
assert list(iter(self.data_array)) == []
self.data_array[(1, 2, 3)] = "12"
self.data_array[(1, 2, 4)] = "13"
assert sorted(list(iter(self.data_array))) == [(1, 2, 3), (1, 2, 4)]
def test_keys(self):
"""Unit test for keys"""
assert self.data_array.keys() == []
self.data_array[(1, 2, 3)] = "12"
self.data_array[(1, 2, 4)] = "13"
assert sorted(self.data_array.keys()) == [(1, 2, 3), (1, 2, 4)]
def test_pop(self):
"""Unit test for pop"""
self.data_array[(1, 2, 3)] = "12"
self.data_array[(1, 2, 4)] = "13"
assert "12" == self.data_array.pop((1, 2, 3))
assert sorted(self.data_array.keys()) == [(1, 2, 4)]
def test_shape(self):
"""Unit test for _get_shape and _set_shape"""
assert self.data_array.shape == (100, 100, 100)
self.data_array.shape = (10000, 100, 100)
assert self.data_array.shape == (10000, 100, 100)
def test_getstate(self):
"""Unit test for __getstate__ (pickle support)"""
assert "dict_grid" in self.data_array.__getstate__()
def test_str(self):
"""Unit test for __str__"""
self.data_array[(1, 2, 3)] = "12"
data_array_str = str(self.data_array)
assert data_array_str == "{(1, 2, 3): '12'}"
def test_slicing(self):
"""Unit test for __getitem__ and __setitem__"""
self.data_array[0, 0, 0] = "'Test'"
self.data_array[0, 0, 0] = "'Tes'"
assert self.data_array[0, 0, 0] == "'Tes'"
def test_cell_array_generator(self):
"""Unit test for cell_array_generator"""
cell_array = self.data_array[:5, 0, 0]
assert list(cell_array) == [None] * 5
cell_array = self.data_array[:5, :5, 0]
assert [list(c) for c in cell_array] == [[None] * 5] * 5
cell_array = self.data_array[:5, :5, :5]
assert [[list(e) for e in c] for c in cell_array] == \
[[[None] * 5] * 5] * 5
def test_adjust_shape(self):
"""Unit test for _adjust_shape"""
self.data_array._adjust_shape(2, 0)
res = list(self.data_array.shape)
assert res == [102, 100, 100]
def test_set_cell_attributes(self):
"""Unit test for _set_cell_attributes"""
cell_attributes = ["Test"]
self.data_array._set_cell_attributes(cell_attributes)
assert self.data_array.cell_attributes == cell_attributes
param_adjust_cell_attributes = [
{'inspoint': 0, 'noins': 5, 'axis': 0,
'src': (4, 3, 0), 'target': (9, 3, 0)},
{'inspoint': 34, 'noins': 5, 'axis': 0,
'src': (4, 3, 0), 'target': (4, 3, 0)},
{'inspoint': 0, 'noins': 0, 'axis': 0,
'src': (4, 3, 0), 'target': (4, 3, 0)},
{'inspoint': 1, 'noins': 5, 'axis': 1,
'src': (4, 3, 0), 'target': (4, 8, 0)},
{'inspoint': 1, 'noins': 5, 'axis': 1,
'src': (4, 3, 1), 'target': (4, 8, 1)},
]
@params(param_adjust_cell_attributes)
def test_adjust_cell_attributes(self, inspoint, noins, axis, src, target):
"""Unit test for _adjust_cell_attributes"""
row, col, tab = src
val = {"angle": 0.2}
attrs = [(Selection([], [], [], [], [(row, col)]), tab, val)]
self.data_array._set_cell_attributes(attrs)
self.data_array._adjust_cell_attributes(inspoint, noins, axis)
for key in val:
assert self.data_array.cell_attributes[target][key] == val[key]
def test_insert(self):
"""Unit test for insert operation"""
self.data_array[2, 3, 4] = 42
self.data_array.insert(1, 100, 0)
assert self.data_array.shape == (200, 100, 100)
assert self.data_array[2, 3, 4] is None
assert self.data_array[102, 3, 4] == 42
def test_delete(self):
"""Tests delete operation"""
self.data_array[2, 3, 4] = "42"
self.data_array.delete(1, 1, 0)
assert self.data_array[2, 3, 4] is None
assert self.data_array[1, 3, 4] == "42"
print self.data_array.shape
assert self.data_array.shape == (99, 100, 100)
try:
self.data_array.delete(1, 1000, 0)
assert False
except ValueError:
pass
def test_set_row_height(self):
"""Unit test for set_row_height"""
self.data_array.set_row_height(7, 1, 22.345)
assert self.data_array.row_heights[7, 1] == 22.345
def test_set_col_width(self):
"""Unit test for set_col_width"""
self.data_array.set_col_width(7, 1, 22.345)
assert self.data_array.col_widths[7, 1] == 22.345
class TestDataArray(object):
"""Unit tests for DataArray"""
def setup_method(self, method):
"""Creates empty DataArray"""
self.data_array = DataArray((100, 100, 100))
def test_iter(self):
"""Unit test for __iter__"""
assert list(iter(self.data_array)) == []
self.data_array[(1, 2, 3)] = "12"
self.data_array[(1, 2, 4)] = "13"
assert sorted(list(iter(self.data_array))) == [(1, 2, 3), (1, 2, 4)]
def test_keys(self):
"""Unit test for keys"""
assert self.data_array.keys() == []
self.data_array[(1, 2, 3)] = "12"
self.data_array[(1, 2, 4)] = "13"
assert sorted(self.data_array.keys()) == [(1, 2, 3), (1, 2, 4)]
def test_pop(self):
"""Unit test for pop"""
self.data_array[(1, 2, 3)] = "12"
self.data_array[(1, 2, 4)] = "13"
assert "12" == self.data_array.pop((1, 2, 3))
assert sorted(self.data_array.keys()) == [(1, 2, 4)]
def test_shape(self):
"""Unit test for _get_shape and _set_shape"""
assert self.data_array.shape == (100, 100, 100)
self.data_array.shape = (10000, 100, 100)
assert self.data_array.shape == (10000, 100, 100)
def test_getstate(self):
"""Unit test for __getstate__ (pickle support)"""
assert "dict_grid" in self.data_array.__getstate__()
def test_str(self):
"""Unit test for __str__"""
self.data_array[(1, 2, 3)] = "12"
data_array_str = str(self.data_array)
assert data_array_str == "{(1, 2, 3): '12'}"
def test_slicing(self):
"""Unit test for __getitem__ and __setitem__"""
self.data_array[0, 0, 0] = "'Test'"
##assert len(self.grid.unredo.undolist) == 1
self.data_array[0, 0, 0] = "'Tes'"
##assert len(self.grid.unredo.undolist) == 2
assert self.data_array[0, 0, 0] == "'Tes'"
def test_cell_array_generator(self):
"""Unit test for cell_array_generator"""
cell_array = self.data_array[:5, 0, 0]
assert list(cell_array) == [None] * 5
cell_array = self.data_array[:5, :5, 0]
assert [list(c) for c in cell_array] == [[None] * 5] * 5
cell_array = self.data_array[:5, :5, :5]
assert [[list(e) for e in c] for c in cell_array] == \
[[[None] * 5] * 5] * 5
def test_adjust_shape(self):
"""Unit test for _adjust_shape"""
self.data_array._adjust_shape(2, 0)
res = list(self.data_array.shape)
assert res == [102, 100, 100]
def test_set_cell_attributes(self):
"""Unit test for _set_cell_attributes"""
pass
def test_adjust_cell_attributes(self):
"""Unit test for _adjust_cell_attributes"""
pass
def test_insert(self):
"""Unit test for insert operation"""
self.data_array[2, 3, 4] = 42
self.data_array.insert(1, 100, 0)
assert self.data_array.shape == (200, 100, 100)
assert self.data_array[2, 3, 4] is None
assert self.data_array[102, 3, 4] == 42
def test_delete(self):
"""Tests delete operation"""
self.data_array[2, 3, 4] = "42"
self.data_array.delete(1, 1, 0)
assert self.data_array[2, 3, 4] is None
assert self.data_array[1, 3, 4] == "42"
print self.data_array.shape
assert self.data_array.shape == (99, 100, 100)
def test_set_row_height(self):
"""Unit test for set_row_height"""
pass
def test_set_col_width(self):
"""Unit test for set_col_width"""
pass
