def setUp(self):
self.test_file_sparse = test_file_sparse = pysal.examples.get_path(
'stata_sparse.txt')
self.test_file_full = test_file_full = pysal.examples.get_path(
'stata_full.txt')
self.obj_sparse = StataTextIO(test_file_sparse, 'r')
self.obj_full = StataTextIO(test_file_full, 'r')
class test_StataTextIO(unittest.TestCase):
def setUp(self):
self.test_file_sparse = test_file_sparse = pysal.examples.get_path(
'stata_sparse.txt')
self.test_file_full = test_file_full = pysal.examples.get_path(
'stata_full.txt')
self.obj_sparse = StataTextIO(test_file_sparse, 'r')
self.obj_full = StataTextIO(test_file_full, 'r')
def test_close(self):
for obj in [self.obj_sparse, self.obj_full]:
f = obj
f.close()
self.assertRaises(ValueError, f.read)
def test_read(self):
w_sparse = self.obj_sparse.read()
self.assertEqual(56, w_sparse.n)
self.assertEqual(4.0, w_sparse.mean_neighbors)
self.assertEqual([1.0, 1.0, 1.0, 1.0, 1.0], list(w_sparse[1].values()))
w_full = self.obj_full.read()
self.assertEqual(56, w_full.n)
self.assertEqual(4.0, w_full.mean_neighbors)
self.assertEqual(
[0.125, 0.125, 0.125, 0.125, 0.125], list(w_full[1].values()))
def test_seek(self):
self.test_read()
self.assertRaises(StopIteration, self.obj_sparse.read)
self.assertRaises(StopIteration, self.obj_full.read)
self.obj_sparse.seek(0)
self.obj_full.seek(0)
self.test_read()
def test_write(self):
for obj in [self.obj_sparse, self.obj_full]:
w = obj.read()
f = tempfile.NamedTemporaryFile(
suffix='.txt', dir=pysal.examples.get_path(''))
fname = f.name
f.close()
o = pysal.open(fname, 'w', 'stata_text')
if obj == self.obj_sparse:
o.write(w)
else:
o.write(w, matrix_form=True)
o.close()
wnew = pysal.open(fname, 'r', 'stata_text').read()
self.assertEqual(wnew.pct_nonzero, w.pct_nonzero)
os.remove(fname)