Fixed linting issues, added SnakeIdioms to test_models.py

3 files changed, 8 insertions, 5 deletions

diff --git a/api/admin.py b/api/admin.py
index de544b52..829d74ed 100644
--- a/api/admin.py
+++ b/api/admin.py
@@ -3,7 +3,7 @@ from django.contrib import admin
 from .models import (
     DocumentationLink, Member,
     OffTopicChannelName, Role,
-    SnakeFact, SnakeIdiom, 
+    SnakeFact, SnakeIdiom,
     SnakeName, Tag
 )
 
diff --git a/api/tests/test_models.py b/api/tests/test_models.py
index 55009bf6..af2ef5d5 100644
--- a/api/tests/test_models.py
+++ b/api/tests/test_models.py
@@ -2,7 +2,9 @@ from django.test import SimpleTestCase
 
 from ..models import (
     DocumentationLink, Member, ModelReprMixin,
-    OffTopicChannelName, Role, SnakeFact, SnakeName, Tag
+    OffTopicChannelName, Role,
+    SnakeFact, SnakeIdiom,
+    SnakeName, Tag
 )
 
 
@@ -28,6 +30,7 @@ class StringDunderMethodTests(SimpleTestCase):
             ),
             OffTopicChannelName(name='bob-the-builders-playground'),
             SnakeFact(fact='snakes are cute'),
+            SnakeIdiom(idiom='snake snacks'),
             SnakeName(name='python', scientific='3'),
             Role(
                 id=5, name='test role',
diff --git a/api/viewsets.py b/api/viewsets.py
index 0025d736..e633c8b7 100644
--- a/api/viewsets.py
+++ b/api/viewsets.py
@@ -11,13 +11,13 @@ from rest_framework_bulk import BulkCreateModelMixin
 
 from .models import (
     DocumentationLink, Member,
-    OffTopicChannelName, SnakeFact, 
+    OffTopicChannelName, SnakeFact,
     SnakeIdiom, SnakeName,
     Tag
 )
 from .serializers import (
     DocumentationLinkSerializer, MemberSerializer,
-    OffTopicChannelNameSerializer, SnakeFactSerializer, 
+    OffTopicChannelNameSerializer, SnakeFactSerializer,
     SnakeIdiomSerializer, SnakeNameSerializer,
     TagSerializer
 )
@@ -212,7 +212,7 @@ class SnakeIdiomViewSet(ListModelMixin, GenericViewSet):
     #### Response format
     >>> [
     ...    {'idiom': 'Sneky snek'},
-    ...    {'idiom': 'Snooky Snake'}     
+    ...    {'idiom': 'Snooky Snake'}
     ... ]
     #### Status codes
     - 200: returned on success