Skip to content

Commit 00b5fab

Browse files
authored
Merge pull request #18 from weikang9009/vec_cor
fix all doctests and require doctests to pass on travis testing
2 parents e3fbbc6 + bc179a1 commit 00b5fab

File tree

3 files changed

+14
-15
lines changed

3 files changed

+14
-15
lines changed

.travis.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ install:
4545
script:
4646
- python setup.py sdist >/dev/null
4747
- python setup.py install
48-
- nosetests --verbose --with-coverage --cover-package=spint;
48+
- nosetests --verbose --with-doctest --with-coverage --cover-package=spint;
4949

5050
notifications:
5151
email:

spint/count_model.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -45,18 +45,18 @@ class CountModel(object):
4545
-------
4646
>>> from spint.count_model import CountModel
4747
>>> import libpysal
48-
>>> db = libpysal.open(libpysal.examples.get_path('columbus.dbf'),'r')
48+
>>> db = libpysal.io.open(libpysal.examples.get_path('columbus.dbf'),'r')
4949
>>> y = np.array(db.by_col("HOVAL"))
5050
>>> y = np.reshape(y, (49,1))
51-
>>> self.y = np.round(y).astype(int)
51+
>>> y = np.round(y).astype(int)
5252
>>> X = []
5353
>>> X.append(db.by_col("INC"))
5454
>>> X.append(db.by_col("CRIME"))
55-
>>> self.X = np.array(X).T
56-
>>> model = CountModel(self.y, self.X, family=Poisson())
55+
>>> X = np.array(X).T
56+
>>> model = CountModel(y, X, family=Poisson())
5757
>>> results = model.fit('GLM')
5858
>>> results.params
59-
array([3.92159085, 0.01183491, -0.01371397])
59+
array([ 3.92159085, 0.01183491, -0.01371397])
6060
6161
"""
6262

spint/gravity.py

+8-9
Original file line numberDiff line numberDiff line change
@@ -145,12 +145,12 @@ class BaseGravity(CountModel):
145145
>>> import numpy as np
146146
>>> import libpysal
147147
>>> from spint.gravity import BaseGravity
148-
>>> db = libpysal.open(libpysal.examples.get_path('nyc_bikes_ct.csv'))
148+
>>> db = libpysal.io.open(libpysal.examples.get_path('nyc_bikes_ct.csv'))
149149
>>> cost = np.array(db.by_col('tripduration')).reshape((-1,1))
150150
>>> flows = np.array(db.by_col('count')).reshape((-1,1))
151151
>>> model = BaseGravity(flows, cost)
152152
>>> model.params
153-
array([ 17.84839637, -1.68325787])
153+
array([17.84839637, -1.68325787])
154154
155155
"""
156156

@@ -431,15 +431,14 @@ class Gravity(BaseGravity):
431431
>>> import numpy as np
432432
>>> import libpysal
433433
>>> from spint.gravity import Gravity
434-
>>> db = libpysal.open(libpysal.examples.get_path('nyc_bikes_ct.csv'))
434+
>>> db = libpysal.io.open(libpysal.examples.get_path('nyc_bikes_ct.csv'))
435435
>>> cost = np.array(db.by_col('tripduration')).reshape((-1,1))
436436
>>> flows = np.array(db.by_col('count')).reshape((-1,1))
437437
>>> o_cap = np.array(db.by_col('o_cap')).reshape((-1,1))
438438
>>> d_cap = np.array(db.by_col('d_cap')).reshape((-1,1))
439439
>>> model = Gravity(flows, o_cap, d_cap, cost, 'exp')
440440
>>> model.params
441-
array([ 3.80050153e+00, 5.54103854e-01, 3.94282921e-01,
442-
-2.27091686e-03])
441+
array([ 3.80050153e+00, 5.54103854e-01, 3.94282921e-01, -2.27091686e-03])
443442
444443
"""
445444

@@ -653,7 +652,7 @@ class Production(BaseGravity):
653652
>>> import numpy as np
654653
>>> import libpysal
655654
>>> from spint.gravity import Production
656-
>>> db = libpysal.open(libpysal.examples.get_path('nyc_bikes_ct.csv'))
655+
>>> db = libpysal.io.open(libpysal.examples.get_path('nyc_bikes_ct.csv'))
657656
>>> cost = np.array(db.by_col('tripduration')).reshape((-1,1))
658657
>>> flows = np.array(db.by_col('count')).reshape((-1,1))
659658
>>> o = np.array(db.by_col('o_tract')).reshape((-1,1))
@@ -867,7 +866,7 @@ class Attraction(BaseGravity):
867866
>>> import numpy as np
868867
>>> import libpysal
869868
>>> from spint.gravity import Attraction
870-
>>> db = libpysal.open(libpysal.examples.get_path('nyc_bikes_ct.csv'))
869+
>>> db = libpysal.io.open(libpysal.examples.get_path('nyc_bikes_ct.csv'))
871870
>>> cost = np.array(db.by_col('tripduration')).reshape((-1,1))
872871
>>> flows = np.array(db.by_col('count')).reshape((-1,1))
873872
>>> d = np.array(db.by_col('d_tract')).reshape((-1,1))
@@ -1082,9 +1081,9 @@ class Doubly(BaseGravity):
10821081
Example
10831082
-------
10841083
>>> import numpy as np
1085-
>>> import libpsal
1084+
>>> import libpysal
10861085
>>> from spint.gravity import Doubly
1087-
>>> db = libpysal.open(libpysal.examples.get_path('nyc_bikes_ct.csv'))
1086+
>>> db = libpysal.io.open(libpysal.examples.get_path('nyc_bikes_ct.csv'))
10881087
>>> cost = np.array(db.by_col('tripduration')).reshape((-1,1))
10891088
>>> flows = np.array(db.by_col('count')).reshape((-1,1))
10901089
>>> d = np.array(db.by_col('d_tract')).reshape((-1,1))

0 commit comments

Comments
 (0)