added tests

Author: glaukiol1

os/os.test.py

@@ -1,23 +1,112 @@
 import os
 
+class bcolors:
+    HEADER = '\033[95m'
+    OKBLUE = '\033[94m'
+    OKCYAN = '\033[96m'
+    OKGREEN = '\033[92m'
+    WARNING = '\033[93m'
+    FAIL = '\033[91m'
+    ENDC = '\033[0m'
+    BOLD = '\033[1m'
+    UNDERLINE = '\033[4m'
+
 print("os.error:", os.error) # prints <class 'OSError'>
-print()
 print("$HOME:", os.environ.get("HOME")) # prints $HOME variable
-print()
 print("cwd():", os.getcwd()) # prints current working directory
-print()
 print("chdir(os.environ.get(\"HOME\"))):", os.chdir(os.environ.get("HOME"))) # changes the current working directory to $HOME
-print()
 print("cwd() after chdir():", os.getcwd()) # should print $HOME dir
-print()
 print("getenv(\"HOME\"):", os.getenv("HOME")) # should print $HOME dir
-print()
 print("getpid():", os.getpid()) # prints the process id
-print()
 print('putenv("TEST_VAR", "TEST_VALUE"):', os.putenv("TEST_VAR", "TEST_VALUE")) # creates a new enviroment variable
-print()
 print('getenv("TEST_VAR"):', os.getenv("TEST_VAR")) # get TEST_VAR value
-print()
 print('unsetenv("TEST_VAR"):', os.unsetenv("TEST_VAR")) # unset TEST_VAR
+print('getenv("TEST_VAR"):',os.getenv("TEST_VAR")) # get TEST_VAR after it has been deleted
+print()
+print("--------------")
+print()
+
+# tests
+passed_tests = 0
+failed = 0
+
+def _pass(test_name):
+    print(bcolors.OKGREEN+test_name+" passed!")
+
+def _fail(test_name):
+    print(bcolors.FAIL+test_name+" failed!")
+
+if os.error == OSError:
+    _pass("os.error == OSError")
+    passed_tests = passed_tests + 1
+else:
+    _fail("os.error == OSError")
+    failed = failed + 1
+
+if os.environ.get("HOME") != None:
+    _pass('os.environ.get("HOME") != None')
+    passed_tests+=1
+else:
+    _fail('os.environ.get("HOME") != None')
+    failed+=1
+
+if os.getcwd() != None:
+    _pass('os.getcwd() != None')
+    passed_tests+=1
+else:
+    _fail('os.getcwd() != None')
+    failed+=1
+
+if os.chdir(os.environ.get("HOME")) == None:
+    _pass('os.chdir(os.environ.get("HOME")) == None')
+    passed_tests+=1
+else:
+    _fail('os.chdir(os.environ.get("HOME")) == None')
+    failed+=1
+
+if os.getcwd() == os.environ.get("HOME"):
+    _pass('os.getcwd() == os.environ.get("HOME")')
+    passed_tests+=1
+else:
+    _fail('os.getcwd() == os.environ.get("HOME")')
+    failed+=1
+
+if os.getpid()>1:
+    _pass('os.getpid() > 1')
+    passed_tests+=1
+else:
+    _fail('os.getpid() > 1')
+    failed+=1
+
+if os.putenv("TEST_VAR", "TEST_VALUE") == None:
+    _pass('os.putenv("TEST_VAR", "TEST_VALUE") == None')
+    passed_tests+=1
+else:
+    _fail('os.putenv("TEST_VAR", "TEST_VALUE") == None')
+    failed+=1
+
+if os.getenv("TEST_VAR") == "TEST_VALUE":
+    _pass('os.getenv("TEST_VAR") == "TEST_VALUE"')
+    passed_tests+=1
+else:
+    _fail('os.getenv("TEST_VAR") == "TEST_VALUE"')
+    failed+=1
+
+if os.unsetenv("TEST_VAR") == None:
+    _pass('os.unsetenv("TEST_VAR") == None')
+    passed_tests+=1
+else:
+    _fail('os.unsetenv("TEST_VAR") == None')
+    failed+=1
+
+if os.getenv("TEST_VAR") == None:
+    _pass('os.getenv("TEST_VAR") == None')
+    passed_tests+=1
+else:
+    _fail('os.getenv("TEST_VAR") == None')
+    failed+=1
+
 print()
-print('getenv("TEST_VAR"):',os.getenv("TEST_VAR")) # get TEST_VAR after it has been deleted
+print(bcolors.ENDC+bcolors.BOLD+bcolors.HEADER+"----- Results -----")
+print(bcolors.OKGREEN+bcolors.BOLD, "PASSED: "+str(passed_tests)+" out of "+str(passed_tests+failed))
+print(bcolors.FAIL+bcolors.BOLD, "FAILED: "+str(failed)+" out of "+str(passed_tests+failed))