Update test_lib.py