added os.getcwdb()

Author: glaukiol1

main

@@ -17,6 +17,7 @@ func init() {
 
 	methods := []*py.Method{
 		py.MustNewMethod("getcwd", getCwd, 0, "Get the current working directory"),
+		py.MustNewMethod("getcwdb", getCwdb, 0, "Get the current working directory in a byte slice"),
 		py.MustNewMethod("chdir", chdir, 0, "Change the current working directory"),
 		py.MustNewMethod("getenv", getenv, 0, "Return the value of the environment variable key if it exists, or default if it doesn’t. key, default and the result are str."),
 		py.MustNewMethod("getpid", getpid, 0, "Return the current process id."),
@@ -62,6 +63,15 @@ func getCwd(self py.Object, args py.Tuple) (py.Object, error) {
 	return py.String(dir), nil
 }
 
+// gets the current working directory in a byte list
+func getCwdb(self py.Object, args py.Tuple) (py.Object, error) {
+	dir, err := os.Getwd()
+	if err != nil {
+		return nil, py.ExceptionNewf(py.OSError, "Unable to get current working directory.")
+	}
+	return py.Bytes([]byte(dir)), nil
+}
+
 // change current working directory
 func chdir(self py.Object, args py.Tuple) (py.Object, error) {
 	if len(args) == 0 {

os/os.module.go

@@ -1,4 +1,5 @@
 import os
+import sys
 import time
 
 class bcolors:
@@ -17,6 +18,7 @@ class bcolors:
 print("os.error:", os.error) # prints <class 'OSError'>
 print("os.environ.get('HOME'):", os.environ.get("HOME")) # prints $HOME variable
 print("cwd():", os.getcwd()) # prints current working directory
+print("cwdb():", os.getcwdb())
 print("chdir(os.environ.get(\"HOME\"))):", os.chdir(os.environ.get("HOME"))) # changes the current working directory to $HOME
 print("cwd() after chdir():", os.getcwd()) # should print $HOME dir
 print("getenv(\"HOME\"):", os.getenv("HOME")) # should print $HOME dir
@@ -34,7 +36,6 @@ class bcolors:
 # tests
 passed_tests = 0
 failed = 0
-
 def _pass(test_name):
     print(bcolors.OKGREEN+test_name+" passed!")
 
@@ -160,18 +161,26 @@ def _fail(test_name):
     _fail('os.getenv("TEST_VAR") == None')
     failed+=1
 
+if bytes(os.getcwd(), "utf-8") == os.getcwdb():
+    _pass('bytes(os.getcwd(), "utf-8") == os.getcwdb()')
+    passed_tests+=1
+else:
+    _fail('bytes(os.getcwd(), "utf-8") == os.getcwdb()')
+    failed+=1
+
 print()
 print(bcolors.ENDC+bcolors.BOLD+bcolors.HEADER+"Please check if $HOME will be outputted...")
 print(bcolors.ENDC)
 
 if os.system("pwd") == 0:
+    print(bcolors.WARNING+'please manually check if your home directory was outputted above, if it isn\'t, this test is a fail! Result by status code: %s' % "pass")
     _pass('os.system("pwd")')
     passed_tests+=1
 else:
+    print(bcolors.WARNING+'please manually check if your home directory was outputted above, if it isn\'t, this test is a fail! Result by status code: %s' % "fail")
     _fail('os.system("pwd")')
     failed+=1
 
-time.sleep(2)
 
 print()
 print(bcolors.ENDC+bcolors.BOLD+bcolors.HEADER+"----- Results -----")