Skip to content

Commit ed89f6b

Browse files
committed
test: kde negated
1 parent 310fb0d commit ed89f6b

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed
Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
datapoint_labels PC1 PC2
2+
test-data-s1 -1.01220906 0.05003395
3+
test-data-s2 -0.84372464 -0.59953316
4+
test-data-s3 1.56185985 -0.48650911
5+
test-data-empty 0.29407385 1.03600832
6+
centroid 0.0 0.0
7+
kde_peak 0.65469949 0.06343901

test/ml/test_ml.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,10 +99,12 @@ def test_pca_kernel_density(self):
9999
OUT_DIR + 'pca-coordinates-kde.tsv', kde=True)
100100
coord = pd.read_table(OUT_DIR + 'pca-coordinates-kde.tsv')
101101
expected = pd.read_table(EXPECT_DIR + 'expected-pca-coordinates-kde.tsv')
102+
expected_negated = pd.read_table(EXPECT_DIR + 'expected-pca-coordinates-kde-negated.tsv')
102103
coord_kde_peak = coord.loc[coord['datapoint_labels'] == 'kde_peak'].round(5)
103104
expected_kde_peak = expected.loc[expected['datapoint_labels'] == 'kde_peak'].round(5)
105+
expected_kde_peak_negated = expected_negated.loc[expected_negated['datapoint_labels'] == 'kde_peak'].round(5)
104106

105-
assert coord_kde_peak.equals(expected_kde_peak)
107+
assert coord_kde_peak.equals(expected_kde_peak) or coord_kde_peak.equals(expected_kde_peak_negated)
106108

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

0 commit comments

Comments
 (0)