def test_multiple(self):
with CodeWriterIndent():
self.assertEqual(CodeWriterIndent.current_indent(), 0)
with CodeWriterIndent(1):
self.assertEqual(CodeWriterIndent.current_indent(), 1)
self.assertEqual(CodeWriterIndent.current_indent(), 0)
with CodeWriterIndent(2) as cwi_2:
self.assertEqual(CodeWriterIndent.current_indent(), 2)
def test_with_code_indent(self):
with open(self.output_file_path, "w", newline="\n") as f:
writer = CodeWriter()
writer.addln("import module_1")
writer.new_line(2)
writer.addln("def add(x1, x2):")
with CodeWriterIndent(1):
writer.addln("pass")
writer.new_line(2)
writer.addln("f: float = 0.5")
writer.format(style_config="pep8")
writer.write(f)
expect_file_path = "{}/{}".format(self.data_dir, "code_writer_test_with_code_indent.py")
actual_file_path = self.output_file_path
with open(actual_file_path, "r") as f:
self.log(f.read())
self.assertTrue(filecmp.cmp(expect_file_path, actual_file_path))
def test_single(self):
with CodeWriterIndent() as cwi:
self.assertEqual(CodeWriterIndent.current_indent(), 0)