from attest import Assert, Tests
import pyhis


twdb = Tests()


@twdb.context
def get_twdb_client():
    source = pyhis.Source(
        'http://his.crwr.utexas.edu/TWDB_Sondes/cuahsi_1_0.asmx?WSDL')
    yield source


@twdb.test
def check_sites(source):
    assert Assert(len(source.sites)) == 74


@twdb.test
def check_variables(source):
    assert Assert(len(source.sites[0].variables)) == 5


if __name__ == '__main__':
    twdb.main()

@@ -26,16 +26,17 @@
 , Tests%0A
+%0A
 import p
@@ -57,16 +57,46 @@
 Tests()%0A
+tceq = Tests()%0Atpwd = Tests()%0A
 %0A%0A@twdb.
@@ -356,32 +356,32 @@
 iables(source):%0A
-
     assert Asser
@@ -421,16 +421,580 @@
  == 5%0A%0A%0A
+@twdb.test%0Adef check_dataframe(source):%0A    df = source.sites%5B0%5D.dataframe%0A    assert Assert(len(df%5B'SAL001'%5D)) == 706%0A%0A%0A@twdb.test%0Adef check_dataframe2(source):%0A    df = source.sites%5B0%5D.dataframe%0A    assert Assert(len(df%5B'SAL001'%5D)) == 706%0A%0A%0A@tceq.context%0Adef get_tceq_client():%0A    source = pyhis.Source(%0A        'http://www.twdb.state.tx.us/appws/histceqswqmis/cuahsi_1_0.asmx?WSDL')%0A    yield source%0A%0A%0A@tceq.test%0Adef check_sites(source):%0A    assert Assert(len(source.sites)) == 8960%0A%0A%0A@tceq.test%0Adef get_dataframe(source):%0A    df = source.sites%5B0%5D.dataframe%0A%0A%0A
 if __nam

add a bunch of tests