[#5] Start adding python unittests