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()