more unit tests

Author: mitzimorris

test/test_optimize.py

@@ -4,6 +4,7 @@
 import io
 import json
 import os
+import shutil
 import unittest
 
 import numpy as np
@@ -582,5 +583,24 @@ def test_show_console(self):
         self.assertTrue('Chain [1] method = optimize' in console)
 
 
+    def test_exe_only(self):
+        stan = os.path.join(DATAFILES_PATH, 'bernoulli.stan')
+        bern_model = CmdStanModel(stan_file=stan)
+        exe_only = os.path.join(DATAFILES_PATH, 'exe_only')
+        shutil.copyfile(bern_model.exe_file, exe_only)
+        os.chmod(exe_only, 0o755)
+
+        bern2_model = CmdStanModel(exe_file=exe_only)
+        jdata = os.path.join(DATAFILES_PATH, 'bernoulli.data.json')
+        mle = bern_model.optimize(data=jdata)
+        self.assertEqual(
+            mle.optimized_params_np[0], mle.optimized_params_dict['lp__']
+        )
+        self.assertEqual(
+            mle.optimized_params_np[1], mle.optimized_params_dict['theta']
+        )
+
+
+
 if __name__ == '__main__':
     unittest.main()

test/test_variational.py

@@ -3,6 +3,7 @@
 import contextlib
 import io
 import os
+import shutil
 import unittest
 from math import fabs
 
@@ -233,5 +234,19 @@ def test_show_console(self):
         self.assertTrue('Chain [1] method = variational' in console)
 
 
+    def test_exe_only(self):
+        stan = os.path.join(DATAFILES_PATH, 'bernoulli.stan')
+        bern_model = CmdStanModel(stan_file=stan)
+        exe_only = os.path.join(DATAFILES_PATH, 'exe_only')
+        shutil.copyfile(bern_model.exe_file, exe_only)
+        os.chmod(exe_only, 0o755)
+
+        bern2_model = CmdStanModel(exe_file=exe_only)
+        jdata = os.path.join(DATAFILES_PATH, 'bernoulli.data.json')
+        variational = bern2_model.variational(data=jdata, algorithm='meanfield')
+        self.assertEqual(variational.variational_sample.shape, (1000, 4))
+
+
+
 if __name__ == '__main__':
     unittest.main()