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)