pub_soft/foldingdiff
3
0
Fork 0
mirror of https://github.com/microsoft/foldingdiff.git synced 2026-06-04 13:30:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix some tests

Browse Source
This commit is contained in:
Kevin Wu
2022-10-31 15:10:20 -07:00
parent c382538645
commit fabfb694cc
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_data.py
View File

@@ -25,7 +25,7 @@ class TestCathCanonical(unittest.TestCase):
def test_return_keys(self):
"""Test that returned dictionary has expected keys"""
d = self.dset[0]
self.assertEqual(set(d.keys()), set(["angles", "coords", "position_ids", "attn_mask"]))
self.assertEqual(set(d.keys()), set(["angles", "coords", "position_ids", "attn_mask", "lengths"]))
def test_num_feature(self):
"""Test that we have the expected number of features"""
@@ -66,7 +66,7 @@ class TestCathCanonicalAnglesOnly(unittest.TestCase):
def test_return_keys(self):
"""Test that returned dictionary has expected keys"""
d = self.dset[0]
self.assertEqual(set(d.keys()), set(["angles", "position_ids", "attn_mask"]))
self.assertEqual(set(d.keys()), set(["angles", "position_ids", "attn_mask", "coords", "lengths"]))
def test_num_features(self):
"""Test that we return the expected number of features and have correctly removed distance"""