def test_mixed_list():
mat = np.random.rand(3, 20)
df = pd.DataFrame(np.random.rand(3, 20))
text = ['here is some test text', 'and a little more', 'and more']
string = 'a string'
res = format_data([mat, df, text, string])
assert isinstance(res, list)
assert all(map(lambda x: isinstance(x, np.ndarray), res))
def test_force_align():
mat = np.random.rand(4, 3)
df = pd.DataFrame(np.random.rand(4, 3))
text = [
'here is some test text', 'and a little more', 'and more',
'just a bit more'
]
res = format_data([mat, df, text])
assert isinstance(res, list)
assert all(map(lambda x: isinstance(x, np.ndarray), res))
assert all(map(lambda x: x.shape[1] == 100, res))
def test_np_array():
data = np.random.rand(100, 10)
assert isinstance(format_data(data), list)
assert isinstance(format_data(data)[0], np.ndarray)
def test_missing_data():
data = np.random.rand(100, 10)
data[0][0] = np.nan
geo = plot(data, show=False)
assert isinstance(format_data(geo), list)
assert isinstance(format_data(geo)[0], np.ndarray)
def test_geo():
geo = plot(np.random.rand(100, 10), show=False)
assert isinstance(format_data(geo), list)
assert isinstance(format_data(geo)[0], np.ndarray)
def test_str():
res = format_data('here is some test text')
assert isinstance(res, list)
assert isinstance(res[0], np.ndarray)
def test_text():
data = ['here is some test text', 'and a little more', 'and more']
assert isinstance(format_data(data), list)
assert isinstance(format_data(data)[0], np.ndarray)
def test_df():
data = pd.DataFrame(np.random.rand(100, 10))
assert isinstance(format_data(data), list)
assert isinstance(format_data(data)[0], np.ndarray)
def test_mixed_list():
data1 = np.random.rand(100, 10)
data2 = pd.DataFrame(np.random.rand(100, 10))
assert isinstance(format_data([data1, data2]), list)
assert isinstance(format_data([data1, data2])[0], np.ndarray)
assert isinstance(format_data([data1, data2])[1], np.ndarray)
