def test_check_variance(): ld = pd.DataFrame(np.vstack((np.ones(3), np.arange(3))).T) M_annot = np.array([[1, 2]]) M_annot, ld, novar_cols = s._check_variance(log, M_annot, ld) assert_array_equal(M_annot.shape, (1, 1)) assert_array_equal(M_annot, [[2]]) assert_array_equal(ld.iloc[:, 0].values, [0.0, 1, 2]) assert_array_equal(set(novar_cols), {True, False})
def test_check_variance(): ld = pd.DataFrame({'SNP': ['a', 'b', 'c'], 'LD1': np.ones(3).astype(float), 'LD2': np.arange(3).astype(float)}) ld = ld[['SNP', 'LD1', 'LD2']] M_annot = np.array([[1, 2]]) M_annot, ld, novar_col = s._check_variance(log, M_annot, ld) assert_array_equal(M_annot.shape, (1, 1)) assert_array_equal(M_annot, [[2]]) assert_allclose(ld.loc[:, 1], [0, 1, 2]) assert_array_equal(novar_col, [True, False])
def test_check_variance(): ld = pd.DataFrame({'SNP': ['a', 'b', 'c'], 'LD1': np.ones(3).astype(float), 'LD2': np.arange(3).astype(float)}) ld = ld[['SNP', 'LD1', 'LD2']] M_annot = np.array([[1, 2]]) M_annot, ld, novar_col = s._check_variance(log, M_annot, ld) assert_array_equal(M_annot.shape, (1, 1)) assert_array_equal(M_annot, [[2]]) assert_allclose(ld.iloc[:, 1], [0, 1, 2]) assert_array_equal(novar_col, [True, False])