class Write():
def __init__(self):
self.pencil = Pencil(pencilLength=4, eraserDurability=10)
self.paper = ''
def write_on_paper(self, desired_text_to_write):
text_to_write = self.pencil.pencil_write(desired_text_to_write)
self.paper += text_to_write
def sharpen_pencil(self):
self.pencil.pencil_sharpen()
def erase(self, desired_text_to_erase):
text_to_erase = self.pencil.pencil_erase(
text_to_erase=desired_text_to_erase)
index = self.paper.rfind(desired_text_to_erase)
self.paper = self.paper[:index] + text_to_erase + self.paper[
index + len(text_to_erase):]
return index
def edit(self, index, text_to_add):
current_text = self.paper[index:index + len(text_to_add)]
for i in range(0, len(text_to_add)):
if current_text[i] != text_to_add[i]:
if not current_text[i].isspace(
) and not text_to_add[i].isspace():
text_to_add = text_to_add[:i] + '@' + text_to_add[i + 1:]
else:
text_to_add = text_to_add[:i] + text_to_add[
i] + text_to_add[i + 1:]
text_to_write = self.pencil.pencil_write(text_to_add)
self.paper = self.paper[:index] + text_to_write + self.paper[
index + len(text_to_add):]
def test_sharpen_pencil_decrease_length_by_one(self):
p = Pencil(pencilLength=4, eraserDurability=4)
new_text = p.pencil_write('abcdef')
self.assertEqual(p.pointDurability, 39994)
p.pencil_sharpen()
self.assertEqual(p.pointDurability, 40000)
self.assertEqual(p.pencilLength, 3)
def test_write_max_point_durability(self):
p = Pencil(pencilLength=4, eraserDurability=4)
f = open('test/input/sample_input.txt', 'r')
input_text = f.read()
f.close()
f = open('test/input/sample_output.txt', 'r')
output_text = f.read()
f.close()
written_text = p.pencil_write(text_to_write=input_text)
self.assertEqual(p.pointDurability, 0)
self.assertEqual(written_text, output_text)
def test_write_one_lowercase_letter_and_one_whitespace_decrease_point_durabitlity_by_one(
self):
p = Pencil(pencilLength=4, eraserDurability=4)
written_text = p.pencil_write(text_to_write="a ")
self.assertEqual(p.pointDurability, 39999)
self.assertEqual(written_text, "a ")
def test_write_two_lowercase_letter_two_uppercase_letter_decrease_point_durabilty_by_six(
self):
p = Pencil(pencilLength=4, eraserDurability=4)
written_text = p.pencil_write(text_to_write="ABcd")
self.assertEqual(p.pointDurability, 39994)
self.assertEqual(written_text, "ABcd")
