add docstrings

0xti4n · Oct 4, 2017 · cb7d5ee · cb7d5ee
1 parent 0b4b704
commit cb7d5ee
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tests/test_models/test_amenity.py b/tests/test_models/test_amenity.py
@@ -61,6 +61,7 @@ def test_amenity_func_docstrings(self):
 class TestAmenity(unittest.TestCase):
  """Test the Amenity class"""
  def test_is_subclass(self):
+ """Test that Amenity is a subclass of BaseModel"""
  amenity = Amenity()
  self.assertIsInstance(amenity, BaseModel)
  self.assertTrue(hasattr(amenity, "id"))

diff --git a/tests/test_models/test_base_model.py b/tests/test_models/test_base_model.py
@@ -1,4 +1,9 @@
 #!/usr/bin/python3
+"""
+Contains class TestBaseModelDocs and class TestBaseModel
+"""
+
+
 from datetime import datetime
 import inspect
 from models import base_model