Practice testing

Author: devanshijbhatt

Library_management.py

@@ -1,5 +1,8 @@
 import pandas as pd
-import datetime
+
+class BookNotAvailableError(Exception):
+    pass
+
 class Book:
     def __init__(self, isbn, title, author, publication_year):
         self.isbn = isbn
@@ -19,14 +22,14 @@ def __init__(self):
         self.books_df = pd.DataFrame(columns=["ISBN", "Title", "Author", "Publication Year", "Is Borrowed"])
 
     def add_book(self, title, author, publication_year):
-        current_year = datetime.datetime.now().year
-        if publication_year > current_year:
-            raise ValueError("The publication year cannot be in the future.")
-        
+        # Increment the ISBN counter by 1 for each new book
         Library_Management.isbn_counter += 1
         isbn = Library_Management.isbn_counter
+        
+        # Create a new Book object
         book = Book(isbn, title, author, publication_year)
 
+        # Add the book details to the DataFrame using concat
         new_book_df = pd.DataFrame([{
             "ISBN": book.isbn,
             "Title": book.title,
@@ -35,6 +38,8 @@ def add_book(self, title, author, publication_year):
             "Is Borrowed": False
         }])
         self.books_df = pd.concat([self.books_df, new_book_df], ignore_index=True)
+        print(f"Book '{book.title}' added to the library with ISBN '{book.isbn}'.")
+        # print(self.books_df)
 
     def borrow_book(self, book_title, isbn):
         # Check if the book is available for borrowing

Test_cases.py

@@ -10,18 +10,19 @@ def test_add_books(self):
 
         # Add a book to the library
         library.add_book("The Great Gatsby", "F. Scott Fitzgerald", 1925)
-        
-        with self.assertRaises(ValueError) as context:
-            library.add_book("The Great Gatsby", "F. Scott Fitzgerald", 2225)
-        
-        # Verify that the correct error message is raised
-        self.assertEqual(str(context.exception), "The publication year cannot be in the future.")
+
         # Check if the book was added successfully
         # added_book = library.books_df.iloc[0]
         # self.assertEqual(added_book["Title"], "The Great Gatsby")
         # self.assertEqual(added_book["Author"], "F. Scott Fitzgerald")
         # self.assertEqual(added_book["Publication Year"], 1925)
-    
+        
+        # library.add_book("To Kill a Mockingbird", "Harper Lee", 1960)
+        # added_book = library.books_df.iloc[1]
+        # self.assertEqual(added_book["Title"], "To Kill a Mockingbird")
+        # self.assertEqual(added_book["Author"], "Harper Lee")
+        # self.assertEqual(added_book["Publication Year"], 1960)
+        
     def test_borrow_book(self):
         """
         Test that it can borrow a book from the library if it's available
@@ -90,7 +91,6 @@ def test_view_available_books(self):
         library.borrow_book("The Twilight Saga",2)
         library.return_book("The Great Gatsby",4)
         library.view_available_books()
-        
 
 if __name__ == '__main__':
     unittest.main()