Skip to content

Commit

Permalink
moved expected
Browse files Browse the repository at this point in the history
  • Loading branch information
ntalluri committed Sep 10, 2024
1 parent e2e4f00 commit 8188a12
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions test/ml/test_ml.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,16 +64,15 @@ def test_pca(self):

def test_pca_robustness(self):
dataframe = ml.summarize_networks([INPUT_DIR + 'test-data-s1/s1.txt', INPUT_DIR + 'test-data-s2/s2.txt', INPUT_DIR + 'test-data-s3/s3.txt'])

expected = pd.read_table(EXPECT_DIR + 'expected-pca-coordinates.tsv')
expected = expected.round(5)
for _ in range(5):
dataframe_shuffled = dataframe.sample(frac=1, axis=1) # permute the columns
ml.pca(dataframe_shuffled, OUT_DIR + 'pca-shuffled-columns.png', OUT_DIR + 'pca-shuffled-columns-variance.txt',
OUT_DIR + 'pca-shuffled-columns-coordinates.tsv')
coord = pd.read_table(OUT_DIR + 'pca-shuffled-columns-coordinates.tsv')
coord = coord.round(5) # round values to 5 digits to account for numeric differences across machines
coord.sort_values(by='algorithm', ignore_index=True, inplace=True)
expected = pd.read_table(EXPECT_DIR + 'expected-pca-coordinates.tsv')
expected = expected.round(5)

assert coord.equals(expected)

Expand All @@ -84,8 +83,6 @@ def test_pca_robustness(self):
coord = pd.read_table(OUT_DIR + 'pca-shuffled-rows-coordinates.tsv')
coord = coord.round(5) # round values to 5 digits to account for numeric differences across machines
coord.sort_values(by='algorithm', ignore_index=True, inplace=True)
expected = pd.read_table(EXPECT_DIR + 'expected-pca-coordinates.tsv')
expected = expected.round(5)

assert coord.equals(expected)

Expand Down

0 comments on commit 8188a12

Please sign in to comment.