Test.py to run test
Added one_hot custom encoder