Update tests

Author: frmdstryr

tests/utils.py renamed to tests/conftest.py

@@ -1,15 +1,9 @@
 import pytest
 from textwrap import dedent
-from utils import compile_source
+from conftest import compile_source
 from web.core.app import WebApplication
 
 
-@pytest.fixture
-def app():
-    app = WebApplication.instance() or WebApplication()
-    yield app
-
-
 def test_block(app):
     # Test that a block's content is replaced
     Page = compile_source(

tests/test_block.py

@@ -1,9 +1,8 @@
-import sys
 import inspect
 import pytest
 from textwrap import dedent
 from lxml import html
-from utils import compile_source, app
+from conftest import compile_source
 
 try:
     import nbformat  # noqa: F401
@@ -19,6 +18,7 @@
 except ImportError:
     MARKDOWN_UNAVAILABLE = True
 
+
 def test_hello_world(app):
     Page = compile_source(
         dedent(

tests/test_html.py

@@ -1,11 +1,11 @@
 import sys
 import time
-import pytest
 import tracemalloc
 import linecache
+import pytest
 from subprocess import Popen, PIPE, STDOUT
 from textwrap import dedent
-from utils import compile_source, app
+from conftest import compile_source
 
 try:
     import requests
@@ -54,29 +54,33 @@ def test_mem_usage():
         print(proc.stdout.read())
 
 
-def diff_stats(snapshot1, snapshot2, label: str, limit: int = 10):
-    top_stats = snapshot2.compare_to(snapshot1, 'lineno')
+def diff_stats(snapshot1, snapshot2, label: str, limit: int = 20):
+    top_stats = snapshot2.compare_to(snapshot1, "lineno")
 
     print(f"------- Top {limit} differences {label} -------")
     for stat in top_stats[:limit]:
         print(stat)
 
 
-def display_top(snapshot, key_type='lineno', limit=10):
-    snapshot = snapshot.filter_traces((
-        tracemalloc.Filter(False, "<frozen importlib._bootstrap>"),
-        tracemalloc.Filter(False, "<unknown>"),
-    ))
+def display_top(snapshot, key_type="lineno", limit=20):
+    snapshot = snapshot.filter_traces(
+        (
+            tracemalloc.Filter(False, "<frozen importlib._bootstrap>"),
+            tracemalloc.Filter(False, "<unknown>"),
+        )
+    )
     top_stats = snapshot.statistics(key_type)
 
     print("Top %s lines" % limit)
     for index, stat in enumerate(top_stats[:limit], 1):
         frame = stat.traceback[0]
-        print("#%s: %s:%s: %.1f KiB (%i)"
-              % (index, frame.filename, frame.lineno, stat.size / 1024, stat.count))
+        print(
+            "#%s: %s:%s: %.1f KiB (%i)"
+            % (index, frame.filename, frame.lineno, stat.size / 1024, stat.count)
+        )
         line = linecache.getline(frame.filename, frame.lineno).strip()
         if line:
-            print('    %s' % line)
+            print("    %s" % line)
 
     other = top_stats[limit:]
     if other:
@@ -86,9 +90,25 @@ def display_top(snapshot, key_type='lineno', limit=10):
     print("Total allocated size: %.1f KiB" % (total / 1024))
 
 
-def test_html_size(app):
+def snap_and_show_stats(snapshots, label):
+    snapshots.append(tracemalloc.take_snapshot())
+    display_top(snapshots[-1])
+    diff_stats(snapshots[-2], snapshots[-1], "after construct")
+
+
+@pytest.fixture
+def tmalloc():
     tracemalloc.start()
-    Page = compile_source(dedent("""
+    try:
+        yield
+    finally:
+        tracemalloc.stop()
+
+
+def test_html_size(tmalloc, app):
+    Page = compile_source(
+        dedent(
+            """
     from web.components.api import *
     from web.core.api import *
     enamldef Page(Html): view:
@@ -108,27 +128,43 @@ def test_html_size(app):
                                 Td:
                                     text = f"{loop.item}"
 
-    """), "Page")
+    """
+        ),
+        "Page",
+    )
+    snapshots = []
+    snapshots.append(tracemalloc.take_snapshot())
+    display_top(snapshots[-1])
 
-    snapshot1 = tracemalloc.take_snapshot()
-    display_top(snapshot1)
+    t0 = time.time()
     view = Page()
-    snapshot2 = tracemalloc.take_snapshot()
-    display_top(snapshot2)
-    diff_stats(snapshot1, snapshot2, "after construct", 20)
 
+    print(f"\n\n\nConstruct took {time.time()-t0}")
+    snap_and_show_stats(snapshots, "after construct")
+
+    t0 = time.time()
+    view.initialize()
+    print(f"\n\n\nInit took {time.time()-t0}")
+    snap_and_show_stats(snapshots, "after init")
+
+    t0 = time.time()
+    view.activate_proxy()
+    print(f"\n\n\nActivate took {time.time()-t0}")
+    snap_and_show_stats(snapshots, "after activate")
+
+    t0 = time.time()
     content = view.render()
-    snapshot3 = tracemalloc.take_snapshot()
-    display_top(snapshot3)
-    diff_stats(snapshot2, snapshot3, "after render", 20)
+    print(f"\n\n\nRender took {time.time()-t0}")
+    snap_and_show_stats(snapshots, "after render")
 
-    assert len(content) < 2_500_000
     total_size = 0
     num_nodes = 0
     for node in view.traverse():
         num_nodes += 1
         total_size += sys.getsizeof(node)
     assert total_size < 5_000_000
+    assert len(content) < 2_500_000
+    assert False
 
 
 if __name__ == "__main__":