adding API and todo to move main() to __main__

random_wc_model_generator/__init__.py

@@ -2,4 +2,9 @@
 
 # read version
 with open(pkg_resources.resource_filename('random_wc_model_generator', 'VERSION'), 'r') as file:
-    __version__ = file.read().strip()
+    __version__ = file.read().strip()
+
+# API
+from .core import CreateWcLangModel, GenerateModel
+from . import enrich_polymers
+from . import random_polymer

random_wc_model_generator/core.py

@@ -142,6 +142,7 @@ def run(self):
         self.summarize_generated_model(wc_lang_model)
         return self
 
+# todo: move to __main__.py
 def main(test_args=None):
     if test_args is None:
         args = GenerateModel.parse_command_line()   # pragma: no cover

tests/test_api.py

@@ -0,0 +1,18 @@
+""" Tests API
+
+:Author: Jonathan Karr <jonrkarr@gmail.com>
+:Date: 2018-03-12
+:Copyright: 2018, Karr Lab
+:License: MIT
+"""
+
+import random_wc_model_generator
+import types
+import unittest
+
+
+class ApiTestCase(unittest.TestCase):
+    def test(self):
+        self.assertIsInstance(random_wc_model_generator, types.ModuleType)
+        self.assertIsInstance(random_wc_model_generator.CreateWcLangModel, type)
+        self.assertIsInstance(random_wc_model_generator.enrich_polymers, types.ModuleType)