Update test_data.py (#292)

5bc96518 · Richard Xue · Gao, Xiang · 027c21b6 · 5bc96518
Commit 5bc96518 authored Aug 15, 2019 by Richard Xue Committed by Gao, Xiang Aug 15, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

tests/test_data.py tests/test_data.py +1 -2

No files found.
--- a/tests/test_data.py
+++ b/tests/test_data.py
@@ -76,8 +76,7 @@ class TestData(unittest.TestCase):
            self.assertLessEqual(species.shape[0], batch_size)
            self.assertEqual(len(coordinates.shape), 3)
            self.assertEqual(coordinates.shape[2], 3)
-            self.assertEqual(coordinates.shape[1], coordinates.shape[1])
-            self.assertEqual(coordinates.shape[0], coordinates.shape[0])
+            self.assertEqual(coordinates.shape[:2], species.shape[:2])
            self.assertEqual(len(energies.shape), 1)
            self.assertEqual(coordinates.shape[0], energies.shape[0])