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)