def test_distance_mean_single_line(self):
with pytest.raises(ValueError) as exc:
ld.distance_mean(self.lines[0][0])
assert exc.value.message == "Lines must contain at least two lines to proceed."
def test_distance_mean_lines_none(self):
with pytest.raises(ValueError) as exc:
ld.distance_mean(None)
assert exc.value.message == "Lines can't be None."
def test_distance_mean_good(self):
result = ld.distance_mean(self.lines)
assert isinstance(result[0], int) and isinstance(result[1], int)