Innocent update to plot.py

.github/workflows/lint.yml

@@ -0,0 +1,51 @@
+name: Pylint
+
+on: 
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  build:
+
+    runs-on: ${{ matrix.os }}
+
+    strategy:
+      matrix:
+        py-version:
+          #- 3.5
+          #- 3.6
+          - 3.7
+          #- 3.8
+          #- 3.9
+        py-arch:
+          - x64
+        mpi:
+          #- mpich
+          - openmpi
+        os:
+          - ubuntu-latest
+          #- ubuntu-18.04
+          #- ubuntu-20.04
+          #- macos-10.15
+          #- macos-latest
+
+    steps:
+    - uses: actions/checkout@v2
+    - uses: jitterbit/get-changed-files@v1
+      id: files
+      with:
+        format: space-delimited
+    - name: Set up Python
+      uses: actions/setup-python@v2
+      with:
+        python-version: ${{ matrix.py-version }}
+        architecture: ${{ matrix.py-arch }}
+    - name: Install dependencies
+      run: python -m pip install pylint
+    - name: Analysing the code with pylint
+      run: |
+        pylint ${{ steps.files.outputs.added_modified }}

.github/workflows/python-package.yml

@@ -0,0 +1,39 @@
+# This workflow will install Python dependencies, run tests and lint with a variety of Python versions
+# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions
+
+name: Python package
+
+on:
+  push:
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
+
+jobs:
+  build:
+
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: [3.7, 3.8, 3.9]
+
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v2
+      with:
+        python-version: ${{ matrix.python-version }}
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        python -m pip install flake8 pytest
+        if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
+    - name: Lint with flake8
+      run: |
+        # stop the build if there are Python syntax errors or undefined names
+        flake8 src --count --select=E9,F63,F7,F82 --show-source --statistics
+        # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
+        flake8 src --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
+    - name: Test with pytest
+      run: |
+        pytest

src/amuse/codes.py

@@ -1,32 +1,34 @@
-
+"""AMUSE codes
+"""
 import sys
 
 _CODES = [
-'athena', 'capreole', 'cachedse',     'gadget2',      'mesa',         'octgrav',      'twobody',
-'capreole',     'hermite0',     'mocassin',     'phiGRAPE',
-'athena',       'evtwin',       'hop',          'seba',
-'bhtree',       'evtwin2sse',   'interface',    'smallN',
-'bse',          'fi',           'mercury',      'sse',
+    'athena', 'capreole', 'cachedse',     'gadget2',      'mesa',
+    'octgrav',      'twobody', 'capreole',     'hermite0',     'mocassin',
+    'phiGRAPE', 'athena',       'evtwin',       'hop',          'seba',
+    'bhtree',       'evtwin2sse',   'interface',    'smallN', 'bse',
+    'fi',           'mercury',      'sse',
 ]
 
 __all__ = []
 
+
 def _import_modules():
 
-    for x in _CODES:
-        modulename = 'amuse.legacy.' + x + '.interface'
+    for code in _CODES:
+        modulename = 'amuse.legacy.' + code + '.interface'
         try:
             __import__(modulename)
-            globals()[x] = sys.modules[modulename]
-            __all__.append(x)
+            globals()[code] = sys.modules[modulename]
+            __all__.append(code)
         except ImportError as ex:
-            modulename = 'amuse.legacy.' + x + '.' + x
+            modulename = 'amuse.legacy.' + code + '.' + code
             try:
                 __import__(modulename) 
-                globals()[x] = sys.modules[modulename]
-                __all__.append(x)
+                globals()[code] = sys.modules[modulename]
+                __all__.append(code)
             except ImportError as ex:
                 pass
 
+
 _import_modules()
-