feat: Implement Python virtual environment display (#137)

tests/testsuite/python.rs

@@ -1,7 +1,9 @@
-use ansi_term::Color;
+use std::env;
 use std::fs::File;
 use std::io;
 
+use ansi_term::Color;
+
 use crate::common;
 
 #[test]
@@ -71,3 +73,23 @@ fn folder_with_py_file() -> io::Result<()> {
     assert_eq!(expected, actual);
     Ok(())
 }
+
+#[test]
+#[ignore]
+fn with_virtual_env() -> io::Result<()> {
+    let dir = common::new_tempdir()?;
+    File::create(dir.path().join("main.py"))?;
+    let output = common::render_module("python")
+        .env("VIRTUAL_ENV", "/foo/bar/my_venv")
+        .arg("--path")
+        .arg(dir.path())
+        .output()?;
+    let actual = String::from_utf8(output.stdout).unwrap();
+
+    let expected = format!(
+        "via {} ",
+        Color::Yellow.bold().paint("🐍 v3.6.9(my_venv)")
+    );
+    assert_eq!(expected, actual);
+    Ok(())
+}