def test_stan_rdump():
data = OrderedDict(x=1, y=0, z=[1, 2, 3], Phi=np.array([[1, 2], [3, 4]]))
tempdir = tempfile.mkdtemp()
filename = os.path.join(tempdir, 'test_rdump.rdump')
misc.stan_rdump(data, filename)
data_recovered = misc.read_rdump(filename)
np.testing.assert_equal(data, data_recovered)
data = OrderedDict(x=True, y=np.array([True, False]))
data_expected = OrderedDict(x=1, y=np.array([1, 0]))
misc.stan_rdump(data, filename)
data_recovered = misc.read_rdump(filename)
np.testing.assert_equal(data_recovered, data_expected)
data = OrderedDict(x='foo')
np.testing.assert_raises(ValueError, misc.stan_rdump, data, filename)
data = OrderedDict(new=3)
np.testing.assert_raises(ValueError, misc.stan_rdump, data, filename)
def test_stan_rdump():
data = OrderedDict(x=1, y=0, z=[1, 2, 3], Phi=np.array([[1, 2], [3, 4]]))
tempdir = tempfile.mkdtemp()
filename = os.path.join(tempdir, 'test_rdump.rdump')
misc.stan_rdump(data, filename)
data_recovered = misc.read_rdump(filename)
np.testing.assert_equal(data, data_recovered)
data = OrderedDict(x=True, y=np.array([True, False]))
data_expected = OrderedDict(x=1, y=np.array([1, 0]))
misc.stan_rdump(data, filename)
data_recovered = misc.read_rdump(filename)
np.testing.assert_equal(data_recovered, data_expected)
data = OrderedDict(x='foo')
np.testing.assert_raises(ValueError, misc.stan_rdump, data, filename)
data = OrderedDict(new=3)
np.testing.assert_raises(ValueError, misc.stan_rdump, data, filename)
def test_rstan_read_rdump():
a = np.array([1, 3, 5])
b = np.arange(10).reshape((-1, 2))
c = np.arange(18).reshape([2, 3, 3])
data = dict(**locals())
tempdir = tempfile.mkdtemp()
filename = os.path.join(tempdir, 'test_rdump.rdump')
misc.stan_rdump(data, filename)
d = misc.read_rdump(filename)
np.testing.assert_equal(d['a'], a)
np.testing.assert_equal(d['b'], b)
np.testing.assert_equal(d['c'], c)
def test_rstan_read_rdump(self):
a = np.array([1, 3, 5])
b = np.arange(10).reshape((-1, 2))
c = np.arange(18).reshape([2, 3, 3])
data = dict(a=a, b=b, c=c)
tempdir = tempfile.mkdtemp()
filename = os.path.join(tempdir, 'test_rdump.rdump')
misc.stan_rdump(data, filename)
d = misc.read_rdump(filename)
np.testing.assert_equal(d['a'], a)
np.testing.assert_equal(d['b'], b)
np.testing.assert_equal(d['c'], c)
def test_stan_read_rdump_array(self):
"""
For reference:
> structure(c(1, 2, 3, 4, 5, 6), .Dim = c(2, 3))
[,1] [,2] [,3]
[1,] 1 3 5
[2,] 2 4 6
"""
# make this into some sort of stream
rdump = 'y <-\nstructure(c(1, 2, 3, 4, 5, 6), .Dim = c(2, 3))\n'
tempdir = tempfile.mkdtemp()
filename = os.path.join(tempdir, 'test_rdump.rdump')
with open(filename, 'w') as f:
f.write(rdump)
d = misc.read_rdump(filename)
y = np.asarray([[1, 3, 5], [2, 4, 6]])
np.testing.assert_equal(d['y'], y)
def test_stan_read_rdump_array(self):
"""
For reference:
> structure(c(1, 2, 3, 4, 5, 6), .Dim = c(2, 3))
[,1] [,2] [,3]
[1,] 1 3 5
[2,] 2 4 6
"""
# make this into some sort of stream
rdump = 'y <-\nstructure(c(1, 2, 3, 4, 5, 6), .Dim = c(2, 3))\n'
tempdir = tempfile.mkdtemp()
filename = os.path.join(tempdir, 'test_rdump.rdump')
with open(filename, 'w') as f:
f.write(rdump)
d = misc.read_rdump(filename)
y = np.asarray([[1, 3, 5], [2, 4, 6]])
np.testing.assert_equal(d['y'], y)
