def test_save(self):
a = AequilibraeMatrix()
a.load(self.sf_skims)
a.computational_view(["distance"])
new_mat = np.random.rand(a.zones, a.zones)
a.matrix_view *= new_mat
res = a.matrix_view.sum()
a.save("new_name_for_matrix")
self.assertEqual(res, a.matrix_view.sum(), "Saved wrong result")
a.save(["new_name_for_matrix2"])
self.assertEqual(a.view_names[0], "new_name_for_matrix2", "Did not update computational view")
self.assertEqual(len(a.view_names), 1, "computational view with the wrong number of matrices")
a.computational_view(["distance", "new_name_for_matrix"])
with self.assertRaises(ValueError):
a.save(["just_one_name"])
a.save(["one_name", "two_names"])
with self.assertRaises(ValueError):
a.save("distance")
b = AequilibraeMatrix()
b.load(self.name_test)
b.computational_view("seed")
b.save()
b.computational_view(["mat", "seed", "dist"])
b.save()
def test_save(self):
a = AequilibraeMatrix()
a.load(self.sf_skims)
a.computational_view(['distance'])
new_mat = np.random.rand(a.zones, a.zones)
a.matrix_view *= new_mat
res = a.matrix_view.sum()
a.save('new_name_for_matrix')
self.assertEqual(res, a.matrix_view.sum(), 'Saved wrong result')
a.save(['new_name_for_matrix2'])
self.assertEqual(a.view_names[0], 'new_name_for_matrix2', 'Did not update computational view')
self.assertEqual(len(a.view_names), 1, 'computational view with the wrong number of matrices')
a.computational_view(['distance', 'new_name_for_matrix'])
with self.assertRaises(ValueError):
a.save(['just_one_name'])
a.save(['one_name', 'two_names'])
with self.assertRaises(ValueError):
a.save('distance')
b = AequilibraeMatrix()
b.load(self.name_test)
b.computational_view("seed")
b.save()
b.computational_view(["mat", "seed", "dist"])
b.save()
