Improving universal selector test

Author: darrenburns

tests/test_query.py

@@ -59,9 +59,9 @@ class App(Widget):
         assert list(app.query("View#main")) == [main_view]
         assert list(app.query("View2#help")) == [help_view]
         assert list(app.query("#widget1")) == [widget1]
-        assert list(app.query("#Widget1")) == [] # Note case.
+        assert list(app.query("#Widget1")) == []  # Note case.
         assert list(app.query("#widget2")) == [widget2]
-        assert list(app.query("#Widget2")) == [] # Note case.
+        assert list(app.query("#Widget2")) == []  # Note case.
 
         assert list(app.query("Widget.float")) == [sidebar, tooltip, helpbar]
         assert list(app.query(Widget).filter(".float")) == [sidebar, tooltip, helpbar]
@@ -131,14 +131,13 @@ class App(Widget):
 
 
 def test_query_classes():
-
     class App(Widget):
         pass
 
     class ClassTest(Widget):
         pass
 
-    CHILD_COUNT=100
+    CHILD_COUNT = 100
 
     # Create a fake app to hold everything else.
     app = App()
@@ -148,34 +147,35 @@ class ClassTest(Widget):
         app._add_child(ClassTest(id=f"child{n}"))
 
     # Let's just be 100% sure everything was created fine.
-    assert len(app.query(ClassTest))==CHILD_COUNT
+    assert len(app.query(ClassTest)) == CHILD_COUNT
 
     # Now, let's check there are *no* children with the test class.
-    assert len(app.query(".test"))==0
+    assert len(app.query(".test")) == 0
 
     # Add the test class to everything and then check again.
     app.query(ClassTest).add_class("test")
-    assert len(app.query(".test"))==CHILD_COUNT
+    assert len(app.query(".test")) == CHILD_COUNT
 
     # Remove the test class from everything then try again.
     app.query(ClassTest).remove_class("test")
-    assert len(app.query(".test"))==0
+    assert len(app.query(".test")) == 0
 
     # Add the test class to everything using set_class.
     app.query(ClassTest).set_class(True, "test")
-    assert len(app.query(".test"))==CHILD_COUNT
+    assert len(app.query(".test")) == CHILD_COUNT
 
     # Remove the test class from everything using set_class.
     app.query(ClassTest).set_class(False, "test")
-    assert len(app.query(".test"))==0
+    assert len(app.query(".test")) == 0
 
     # Add the test class to everything using toggle_class.
     app.query(ClassTest).toggle_class("test")
-    assert len(app.query(".test"))==CHILD_COUNT
+    assert len(app.query(".test")) == CHILD_COUNT
 
     # Remove the test class from everything using toggle_class.
     app.query(ClassTest).toggle_class("test")
-    assert len(app.query(".test"))==0
+    assert len(app.query(".test")) == 0
+
 
 def test_invalid_query():
     class App(Widget):
@@ -210,3 +210,4 @@ def compose(self) -> ComposeResult:
         query = container.query("*")
         results = list(query.results())
         assert len(results) == 5
+        assert not any(node.id == "root-container" for node in results)