read_dataset has at least two dimensions

This commit is contained in:
Martin Diehl 2019-11-23 19:21:41 +01:00
parent eb80a82e05
commit 3be1caa4ef
1 changed files with 6 additions and 6 deletions

View File

@ -51,17 +51,17 @@ class TestDADF5:
assert np.allclose(in_memory,in_file) assert np.allclose(in_memory,in_file)
def test_add_determinant(self,default): def test_add_determinant(self,default):
default.add_determinant('Lp') default.add_determinant('P')
loc = {'Lp': default.get_dataset_location('Lp'), loc = {'P': default.get_dataset_location('P'),
'det(Lp)': default.get_dataset_location('det(Lp)')} 'det(P)': default.get_dataset_location('det(P)')}
in_memory = np.linalg.det(default.read_dataset(loc['Lp'],0)) in_memory = np.linalg.det(default.read_dataset(loc['P'],0)).reshape(-1,1)
in_file = default.read_dataset(loc['det(Lp)'],0) in_file = default.read_dataset(loc['det(P)'],0)
assert np.allclose(in_memory,in_file) assert np.allclose(in_memory,in_file)
def test_add_spherical(self,default): def test_add_spherical(self,default):
default.add_spherical('P') default.add_spherical('P')
loc = {'P': default.get_dataset_location('P'), loc = {'P': default.get_dataset_location('P'),
'p_P': default.get_dataset_location('p_P')} 'p_P': default.get_dataset_location('p_P')}
in_memory = mechanics.spherical_part(default.read_dataset(loc['P'],0)) in_memory = mechanics.spherical_part(default.read_dataset(loc['P'],0)).reshape(-1,1)
in_file = default.read_dataset(loc['p_P'],0) in_file = default.read_dataset(loc['p_P'],0)
assert np.allclose(in_memory,in_file) assert np.allclose(in_memory,in_file)