class test_MtxIO(unittest.TestCase): def setUp(self): self.test_file = test_file = pysal.examples.get_path('wmat.mtx') self.obj = MtxIO(test_file, 'r') def test_close(self): f = self.obj f.close() self.assertRaises(ValueError, f.read) def test_read(self): w = self.obj.read() self.assertEqual(49, w.n) self.assertEqual(4.7346938775510203, w.mean_neighbors) self.assertEqual( [0.33329999999999999, 0.33329999999999999, 0.33329999999999999], list(w[1].values())) s0 = w.s0 self.obj.seek(0) wsp = self.obj.read(sparse=True) self.assertEqual(49, wsp.n) self.assertEqual(s0, wsp.s0) def test_seek(self): self.test_read() self.assertRaises(StopIteration, self.obj.read) self.obj.seek(0) self.test_read() def test_write(self): for i in [False, True]: self.obj.seek(0) w = self.obj.read(sparse=i) f = tempfile.NamedTemporaryFile(suffix='.mtx', dir=pysal.examples.get_path('')) fname = f.name f.close() o = pysal.open(fname, 'w') o.write(w) o.close() wnew = pysal.open(fname, 'r').read(sparse=i) if i: self.assertEqual(wnew.s0, w.s0) else: self.assertEqual(wnew.pct_nonzero, w.pct_nonzero) os.remove(fname)
class test_MtxIO(unittest.TestCase): def setUp(self): self.test_file = test_file = pysal.examples.get_path("wmat.mtx") self.obj = MtxIO(test_file, "r") def test_close(self): f = self.obj f.close() self.failUnlessRaises(ValueError, f.read) def test_read(self): w = self.obj.read() self.assertEqual(49, w.n) self.assertEqual(4.7346938775510203, w.mean_neighbors) self.assertEqual([0.33329999999999999, 0.33329999999999999, 0.33329999999999999], w[1].values()) s0 = w.s0 self.obj.seek(0) wsp = self.obj.read(sparse=True) self.assertEqual(49, wsp.n) self.assertEqual(s0, wsp.s0) def test_seek(self): self.test_read() self.failUnlessRaises(StopIteration, self.obj.read) self.obj.seek(0) self.test_read() def test_write(self): for i in [False, True]: self.obj.seek(0) w = self.obj.read(sparse=i) f = tempfile.NamedTemporaryFile(suffix=".mtx", dir=pysal.examples.get_path("")) fname = f.name f.close() o = pysal.open(fname, "w") o.write(w) o.close() wnew = pysal.open(fname, "r").read(sparse=i) if i: self.assertEqual(wnew.s0, w.s0) else: self.assertEqual(wnew.pct_nonzero, w.pct_nonzero) os.remove(fname)
def setUp(self): self.test_file = test_file = pysal.examples.get_path('wmat.mtx') self.obj = MtxIO(test_file, 'r')
def setUp(self): self.test_file = test_file = pysal.examples.get_path("wmat.mtx") self.obj = MtxIO(test_file, "r")