shapes should also match

This commit is contained in:
Martin Diehl 2020-01-26 10:01:00 +01:00
parent e6cf75ca79
commit 7f92418fc4
1 changed files with 2 additions and 2 deletions

View File

@ -38,14 +38,14 @@ class TestTable:
def test_write_read_str(self,default,tmpdir): def test_write_read_str(self,default,tmpdir):
default.to_ASCII(str(tmpdir.join('default.txt'))) default.to_ASCII(str(tmpdir.join('default.txt')))
new = Table.from_ASCII(str(tmpdir.join('default.txt'))) new = Table.from_ASCII(str(tmpdir.join('default.txt')))
assert all(default.data==new.data) assert all(default.data==new.data) and default.shapes == new.shapes
def test_write_read_file(self,default,tmpdir): def test_write_read_file(self,default,tmpdir):
with open(tmpdir.join('default.txt'),'w') as f: with open(tmpdir.join('default.txt'),'w') as f:
default.to_ASCII(f) default.to_ASCII(f)
with open(tmpdir.join('default.txt')) as f: with open(tmpdir.join('default.txt')) as f:
new = Table.from_ASCII(f) new = Table.from_ASCII(f)
assert all(default.data==new.data) assert all(default.data==new.data) and default.shapes == new.shapes
def test_read_ang_str(self,reference_dir): def test_read_ang_str(self,reference_dir):
new = Table.from_ang(os.path.join(reference_dir,'simple.ang')) new = Table.from_ang(os.path.join(reference_dir,'simple.ang'))