def test_iadd_matrix(self):
"""Test in place add"""
r2 = Matrix(self.data)
r3 = Matrix(self.content)
r2.row += r3
assert r2.row[3] == ['r', 's', 't', 'o', '']
assert r2.row[4] == [1, 2, 3, 4, '']
assert r2.row[5] == [True, "", "", "", '']
assert r2.row[6] == [1.1, 2.2, 3.3, 4.4, 5.5]
def test_iadd_matrix(self):
"""Test in place add"""
r2 = Matrix(self.data)
r3 = Matrix(self.content)
r2.row += r3
assert r2.row[3] == ["r", "s", "t", "o", ""]
assert r2.row[4] == [1, 2, 3, 4, ""]
assert r2.row[5] == [True, "", "", "", ""]
assert r2.row[6] == [1.1, 2.2, 3.3, 4.4, 5.5]
def test_iadd_matrix(self):
"""Test in place add a matrix"""
m5 = Matrix(copy.deepcopy(self.data))
m6 = Matrix(copy.deepcopy(self.data))
m7 = m5.column + m6
result2 = [[1, 2, 3, 4, 5, 6, 1, 2, 3, 4, 5, 6],
[1, 2, 3, 4, '', '', 1, 2, 3, 4, '', ''],
[1, '', '', '', '', '', 1, '', '', '', '', '']]
eq_(result2, m7.get_internal_array())
def test_extend_rows(self):
r = Matrix(self.data)
r.extend_rows(self.content)
assert r.row[3] == ["r", "s", "t", "o", ""]
assert r.row[4] == [1, 2, 3, 4, ""]
assert r.row[5] == [True, "", "", "", ""]
assert r.row[6] == [1.1, 2.2, 3.3, 4.4, 5.5]
def test_add(self):
"""Test operator add overload
"""
# +
m3 = Matrix(self.data)
m4 = m3.column + self.data3
eq_(self.result, m4.get_internal_array())
def test_a_wrong_slice(self):
r = Matrix(self.data)
try:
r.row[2:1] = ['e', 'r', 'r', 'o']
assert 1 == 2
except ValueError:
assert 1 == 1
def test_delete_an_invalid_slice(self):
m = Matrix(self.data)
try:
del m.row[2:1]
assert 1 == 2
except ValueError:
assert 1 == 1
def test_delete_a_index(self):
"""Delete a index"""
r = Matrix(self.data)
content = ['i', 'j', 1.1, 1]
assert r.row[2] == content
del r.row[0]
assert r.row[1] == content
def test_add(self):
r3 = Matrix(self.data)
r3 = r3.row + self.content
assert r3.row[3] == ['r', 's', 't', 'o', '']
assert r3.row[4] == [1, 2, 3, 4, '']
assert r3.row[5] == [True, "", "", "", '']
assert r3.row[6] == [1.1, 2.2, 3.3, 4.4, 5.5]
def test_extend_rows(self):
r = Matrix(self.data)
r.extend_rows(self.content)
assert r.row[3] == ['r', 's', 't', 'o', '']
assert r.row[4] == [1, 2, 3, 4, '']
assert r.row[5] == [True, "", "", "", '']
assert r.row[6] == [1.1, 2.2, 3.3, 4.4, 5.5]
def test_set_an_invalid_slice(self):
m = Matrix(self.data)
try:
m.column[2:1] = ['e', 'r', 'r', 'o']
assert 1 == 2
except ValueError:
assert 1 == 1
def test_a_wrong_slice(self):
r = Matrix(self.data)
try:
r.row[2:1] = ["e", "r", "r", "o"]
assert 1 == 2
except ValueError:
assert 1 == 1
def test_add(self):
r3 = Matrix(self.data)
r3 = r3.row + self.content
assert r3.row[3] == ["r", "s", "t", "o", ""]
assert r3.row[4] == [1, 2, 3, 4, ""]
assert r3.row[5] == [True, "", "", "", ""]
assert r3.row[6] == [1.1, 2.2, 3.3, 4.4, 5.5]
def test_set_column_at(self):
r = Matrix(self.data)
try:
r.set_column_at(1, [11, 1], 1000)
assert 1 == 2
except IndexError:
assert 1 == 1
def test_set_columns(self):
r = Matrix(self.data)
content = ['r', 's', 't', 'o']
r.column[1] = content
assert r.column[1] == content
assert r.column[0] == [1, 1, 1, '']
r.column['B'] = ['p', 'q', 'r']
assert r.column['B'] == ['p', 'q', 'r', 'o']
def test_to_array(self):
m = Matrix(self.data)
result = [
[1, 2, 3, 4, 5, 6],
[1, 2, 3, 4, "", ""],
[1, "", "", "", "", ""],
]
eq_(result, m.get_internal_array())
def test_set_columns(self):
matrix = Matrix(self.data)
content = ["r", "s", "t", "o"]
matrix.column[1] = content
assert matrix.column[1] == content
assert matrix.column[0] == [1, 1, 1, ""]
matrix.column["B"] = ["p", "q", "r"]
eq_(matrix.column["B"], ["p", "q", "r", "o"])
def test_update_a_cell(self):
r = Matrix(self.data)
r[0, 0] = 'k'
assert r[0, 0] == 'k'
d = datetime.date(2014, 10, 1)
r.cell_value(0, 1, d)
assert isinstance(r[0, 1], datetime.date) is True
assert r[0, 1].strftime("%d/%m/%y") == "01/10/14"
r["A1"] = 'p'
assert r[0, 0] == 'p'
r[0, 1] = 16
assert r["B1"] == 16
def setUp(self):
"""
Make a test csv file as:
1,2,3,4
5,6,7,8
9,10,11,12
"""
self.array = []
for i in [0, 4, 8]:
array = [i + 1, i + 2, i + 3, i + 4]
self.array.append(array)
self.iteratable = Matrix(self.array)
def test_wrong_index_type(self):
r = Matrix(self.data)
r["string"][0] # bang, cannot get
def test_iadd_list(self):
"""Test in place add a list
"""
m2 = Matrix(self.data)
m2.column += self.data3
eq_(self.result, m2.get_internal_array())
def test_extend_columns2(self):
"""Test extend columns"""
m = Matrix(self.data)
m.extend_columns(1.1)
def test_extend_column(self):
"""test extend just one column"""
m = Matrix(self.data)
m.extend_columns([1, 1])
assert m.row[0] == self.result[0][:7]
def test_extend_columns(self):
"""Test extend columns"""
m = Matrix(self.data)
m.extend_columns(self.data3)
eq_(self.result, m.get_internal_array())
def test_set_with_a_wrong_index(self):
"""Get with a wrong index"""
m = Matrix(self.data)
m[1.1] = 1 # bang,
def test_set_column_with_a_wrong_column_index(self):
"""Get with a wrong index"""
m = Matrix(self.data)
m.column[1.11] = 1 # bang, string type index
def test_delete_with_a_wrong_column_index(self):
"""Get with a wrong index"""
m = Matrix(self.data)
del m.column[1.11] # bang, string type index
def test_delete_rows_with_invalid_list(self):
m = Matrix([])
m.delete_rows('ab') # bang, cannot delete
def test_wrong_index_type2(self):
r = Matrix(self.data)
r["string"] = 'k' # bang, cannot set
def test_transpose(self):
data = [[1, 2, 3], [4, 5, 6]]
result = [[1, 4], [2, 5], [3, 6]]
m = Matrix(data)
m.transpose()
eq_(result, m.get_internal_array())
