Add pystencils.autodiff

pystencils/__init__.py

@@ -13,6 +13,16 @@ from .simp import AssignmentCollection
 from .slicing import make_slice
 from .sympyextensions import SymbolCreator
 
+try:
+    import sys
+    import pystencils_autodiff
+    sys.modules['pystencils.autodiff'] = pystencils_autodiff
+except Exception:
+    class autodiff:  # NOQA
+        def __getattribute__(self):
+            raise NotImplementedError("pystencils_autodiff is not installed! Run 'pip install pystencils_autodiff'")
+
+
 __all__ = ['Field', 'FieldType', 'fields',
            'TypedSymbol',
            'make_slice',

pystencils_tests/test_import_autodiff.py

+# -*- coding: utf-8 -*-
+#
+# Copyright © 2019 Stephan Seitz <stephan.seitz@fau.de>
+#
+# Distributed under terms of the GPLv3 license.
+
+"""
+
+"""
+
+
+def test_import_autodiff():
+    import pystencils.autodiff
+
+
+def main():
+    test_import_autodiff()
+
+
+if __name__ == '__main__':
+    main()