def testSetDef(self): self.assertEquals(sesame.endpoints["cds"], sesame.default_endpoint) # pdb.set_trace() sesame.set_default_endpoint("cds") self.assertEquals(sesame.endpoints["cds"], sesame.default_endpoint) sesame.set_default_endpoint("cfa") self.assertEquals(sesame.endpoints["cfa"], sesame.default_endpoint)
import os, sys, shutil, re, imp import unittest, pdb from urllib2 import URLError, HTTPError import pyvo.nameresolver.sesame as sesame import xml.etree.ElementTree as ET from astropy.tests.helper import pytest, remote_data mirrors = ["cfa", "cds"] default_mirror = "cds" if default_mirror in mirrors: mirrors.remove(default_mirror) alt_mirror = mirrors[0] sesame.set_default_endpoint(default_mirror) do_remote = False @remote_data class SesameQueryTest(unittest.TestCase): def testExecuteStream(self): q = sesame.SesameQuery() q.names = ["ngc4258", "goob's star", "m51"] strm = q.execute_stream() doc = ET.parse(strm) strm.close() self.assertEquals("Sesame", doc.getroot().tag) strm = q.execute_stream(astext=True)