Skip to content

Commit 3f83637

Browse files
danhphantwiecki
authored andcommitted
add a test for get_data function
1 parent 5588858 commit 3f83637

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

pymc/tests/test_data_container.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,8 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
import io
16+
1517
import numpy as np
1618
import pandas as pd
1719
import pytest
@@ -403,3 +405,8 @@ def test_data_naming():
403405
y = pm.Normal("y")
404406
assert y.name == "named_model_y"
405407
assert x.name == "named_model_x"
408+
409+
410+
def test_get_data():
411+
data = pm.get_data("radon.csv")
412+
assert type(data) == io.BytesIO

0 commit comments

Comments
 (0)