def test_table_margins():
cont_table = [[10, 10, 20], [20, 20, 10]]
t = table_margins(cont_table)
assert t[0][0][0] == 40
cont_table2 = np.array([[[10, 10, 20], [20, 20, 10]]])
with pytest.raises(ValueError):
table_margins(cont_table2)
cont_table3 = np.array([10, 10, 20])
t2 = table_margins(cont_table3)
assert all(t2[0] == cont_table3)
def test_margins_exceptions(self):
with pytest.raises(ValueError):
table_margins(self.cont_table2)
def test_table_margins(self):
t = table_margins(self.cont_table)
t2 = table_margins(self.cont_table3)
assert t[0][0][0] == 40
assert all(t2[0] == self.cont_table3)