def test_no_command(): temp = utils.tempdir() src_path = temp / "src.sh" dest_path = temp / "dest.py" with open(src_path, "w") as src_f: src_f.write("# start\n") src_f.write("# description\n") src_f.write("end\n") bash_to_python(src_path, dest_path) with open(dest_path, "r") as dest_f: generated_cmd = dest_f.read() expected_cmd = "# start\n" "# description\n" "end\n" assert generated_cmd == expected_cmd
def test_last_line_break(): """Test that line endings are correct""" temp = utils.tempdir() src_path = temp / "src.sh" dest_path = temp / "dest.py" with open(src_path, "w") as src_f: src_f.write("# start\n") src_f.write("# end\n") bash_to_python(src_path, dest_path) with open(dest_path, "r") as dest_f: generated_cmd = dest_f.read() expected_cmd = "# start\n" "# end\n" assert generated_cmd == expected_cmd
def test_bash_cmd(): temp = utils.tempdir() src_path = temp / "src.sh" dest_path = temp / "dest.py" with open(src_path, "w") as src_f: src_f.write(BASH) src_f.write("\n") src_f.write("tvmc\n") src_f.write(BASH) src_f.write("\n") bash_to_python(src_path, dest_path) with open(dest_path, "r") as dest_f: generated_cmd = dest_f.read() expected_cmd = "# .. code-block:: bash\n" "#\n" "#\t tvmc\n" "#\n" assert generated_cmd == expected_cmd
def test_multiline_comment(): temp = utils.tempdir() src_path = temp / "src.sh" dest_path = temp / "dest.py" with open(src_path, "w") as src_f: src_f.write(BASH_MULTILINE_COMMENT_START) src_f.write("\n") src_f.write("comment\n") src_f.write(BASH_MULTILINE_COMMENT_END) src_f.write("\n") bash_to_python(src_path, dest_path) with open(dest_path, "r") as dest_f: generated_cmd = dest_f.read() expected_cmd = '"""\n' "comment\n" '"""\n' assert generated_cmd == expected_cmd
def test_bash_ignore_cmd(): temp = utils.tempdir() src_path = temp / "src.sh" dest_path = temp / "dest.py" with open(src_path, "w") as src_f: src_f.write("# start\n") src_f.write(BASH_IGNORE) src_f.write("\n") src_f.write("tvmc\n") src_f.write(BASH_IGNORE) src_f.write("\n") src_f.write("# end\n") bash_to_python(src_path, dest_path) with open(dest_path, "r") as dest_f: generated_cmd = dest_f.read() expected_cmd = "# start\n" "# end\n" assert generated_cmd == expected_cmd
def test_bash_ignore_cmd(): """Test that ignored bash commands are not turned into code blocks""" temp = utils.tempdir() src_path = temp / "src.sh" dest_path = temp / "dest.py" with open(src_path, "w") as src_f: src_f.write("# start\n") src_f.write(BASH_IGNORE) src_f.write("\n") src_f.write("tvmc\n") src_f.write(BASH_IGNORE) src_f.write("\n") src_f.write("# end\n") bash_to_python(src_path, dest_path) with open(dest_path, "r") as dest_f: generated_cmd = dest_f.read() expected_cmd = "# start\n" "# end\n" assert generated_cmd == expected_cmd
def test_text_and_bash_command(): """Test a file with a bash code block""" temp = utils.tempdir() src_path = temp / "src.sh" dest_path = temp / "dest.py" with open(src_path, "w") as src_f: src_f.write("# start\n") src_f.write(BASH) src_f.write("\n") src_f.write("tvmc\n") src_f.write(BASH) src_f.write("\n") src_f.write("# end\n") bash_to_python(src_path, dest_path) with open(dest_path, "r") as dest_f: generated_cmd = dest_f.read() expected_cmd = "# start\n" "# .. code-block:: bash\n" "#\n" "#\t tvmc\n" "#\n" "# end\n" assert generated_cmd == expected_cmd