import numpy as np
fixed the tests