from solution import Book, Library, Student, Classroom
def test_library():
print("Running Library tests...")
lib = Library()
b1 = Book(1, "Python")
b2 = Book(2, "DSA")
assert lib.add_book(b1) is True
assert lib.add_book(b2) is True
assert lib.add_book(b1) is False # duplicate
assert lib.borrow_book(1) is True
assert lib.borrow_book(1) is False
assert lib.return_book(1) is True
assert lib.return_book(1) is False
assert lib.borrow_book(99) is False # invalid ID
print("✅ Library tests passed")
def test_classroom():
print("Running Classroom tests...")
room = Classroom()
s1 = Student(1, "Amit")
s2 = Student(2, "Neha")
assert room.add_student(s1) is True
assert room.add_student(s2) is True
assert room.add_student(s1) is False # duplicate
assert room.get_present_count() == 0
assert room.mark_attendance(1, True) is True
assert room.mark_attendance(1, True) is False
assert room.get_present_count() == 1
assert room.mark_attendance(2, True) is True
assert room.get_present_count() == 2
assert room.mark_attendance(2, False) is True
assert room.get_present_count() == 1
assert room.mark_attendance(99, True) is False
print("✅ Classroom tests passed")
if __name__ == "__main__":
test_library()
test_classroom()
print("\n🎉 ALL DAY 3 TESTS PASSED SUCCESSFULLY")