surimi 0.0.0__py3-none-any.whl

surimi/__init__.py

@@ -0,0 +1,3 @@
+from .decorators import proxy_test_decorator, Test, test
+from .registration import discover_tests
+from .formater import format_failed

surimi/cli.py

@@ -0,0 +1,20 @@
+import click
+import os
+import sys
+from .registration import discover_tests
+from .runner import run_tests
+
+@click.command()
+@click.option('-v', '--verbose', default=0, nargs=1, type=int)
+@click.argument('directory', default='.', nargs=1, type=str)
+def test(verbose, directory):
+  root = os.path.abspath(directory)
+  sys.path.insert(0, root)
+  import surimi
+  tests = surimi.Test()
+  surimi.test = tests
+  discover_tests(directory)
+  run_tests(tests, verbose=verbose)
+
+if __name__ == '__main__':
+  test()

surimi/data.py

@@ -0,0 +1,13 @@
+from dataclasses import dataclass
+from typing import Any
+
+@dataclass
+class TestResult:
+  name: str
+  fail_with: Any
+  location: str
+  failed: bool
+  stdout: str
+  stderr: str
+  execution_time: float = 0.0
+  exception: Exception = None

surimi/decorators/__init__.py

@@ -0,0 +1,3 @@
+from .test import Test, proxy_test_decorator, TestInfos
+
+test = proxy_test_decorator

surimi/decorators/test.py

@@ -0,0 +1,47 @@
+from rich.console import Console
+from dataclasses import dataclass
+from typing import Any, Callable
+
+@dataclass
+class TestInfos:
+  function: Callable[[], None]
+  name: str
+  location: str
+  fail_with: Exception
+
+  def __call__(self):
+    self.function()
+
+class Test:
+  def __init__(self):
+    self.registered_tests: list[TestInfos] = {}
+  
+  def add_function(self, fun, **kwargs):
+      fail_with = kwargs.get('fail_with')
+      test = TestInfos(fun, fun.__name__, fun.__globals__['__file__'], fail_with)
+      module_name = fun.__globals__['__name__']
+      if module_name not in self.registered_tests:
+        self.registered_tests[module_name] = []
+      self.registered_tests[module_name].append(test)
+  
+  def __call__(self, *args, **kwargs):
+    #@test
+    if args and callable(args[0]):
+      self.add_function(args[0], **kwargs)
+      return args[0]
+
+    #@test(...)
+    def register(function):
+      self.add_function(function, **kwargs)
+      return function
+
+    return register
+
+def proxy_test_decorator(*args, **kwargs):
+  if args and callable(args[0]):
+    function = args[0]
+    return function
+
+  def register(function):
+    return function
+  return register

surimi/formater.py

@@ -0,0 +1,39 @@
+import traceback
+from rich.console import Console
+from rich.panel import Panel
+from rich.syntax import Syntax
+from rich.text import Text
+from .data import TestResult
+
+def format_module_result(console: Console, num_module: int, total_modules: int, module: str, failed: bool, execution_time: float):
+  if not failed : status = Text("[bold][green]PASSED[/green][/bold]")
+  else: status = Text("[bold][red]FAILED[/red][/bold]")
+  time_str = f"{execution_time:.2f}s"
+  module = module.replace('.', '/')
+  console.print(f"{num_module}/{total_modules}  {module:<35}  {status}  {time_str:>8}")
+
+def format_failed(console: Console, test_result: TestResult, verbose: int):
+  error_name = type(test_result.exception).__name__
+  error = f'[red]{error_name}[/red]' if test_result.exception else "[green]No error[/green]"
+  expected = f', expected [red]{test_result.fail_with.__name__}[/red]' if test_result.fail_with else ''
+  
+  console.print(f"  -> [red]Test [bold]{test_result.name}[/bold] failed[/red]: got {error}{expected}")
+  
+  if verbose >= 1:
+    if test_result.stdout:
+      console.print(f"     [yellow]stdout:[/yellow]")
+      for line in test_result.stdout.strip().split('\n'):
+        console.print(f"       {line}")
+    
+    if test_result.stderr:
+      console.print(f"     [yellow]stderr:[/yellow]")
+      for line in test_result.stderr.strip().split('\n'):
+        console.print(f"       {line}")
+  
+  if verbose >= 2:
+    if test_result.exception and test_result.exception.__traceback__:
+      console.print(f"     [yellow]traceback:[/yellow]")
+      tb_lines = traceback.format_tb(test_result.exception.__traceback__)
+      for line in tb_lines:
+        for tb_line in line.strip().split('\n'):
+          console.print(f"       {tb_line}")

surimi/registration.py

@@ -0,0 +1,32 @@
+import importlib
+import sys
+import os
+from pathlib import Path 
+
+def discover_tests(dir: str = '.'):
+  test_locations = [
+      dir,
+      os.path.join(dir, 'src'),
+      os.path.join(dir, 'tests'),
+  ]
+  
+  for test_location in test_locations:
+    if not os.path.isdir(test_location):
+      continue
+    
+    for root, dirs, files in os.walk(test_location):
+      dirs[:] = [
+          d for d in dirs 
+          if os.path.exists(os.path.join(root, d, "__init__.py"))
+      ]
+      
+      for file in files:
+        if file.endswith(".py"):
+          rel_path = os.path.relpath(os.path.join(root, file), dir)
+          module_name = rel_path.replace(os.sep, ".").replace(".py", "")
+          
+          try:
+            importlib.import_module(module_name)
+          except ImportError as e:
+            print(f"[red]Error importing {module_name}: {e}[/red]")
+

surimi/runner.py

@@ -0,0 +1,70 @@
+from dataclasses import dataclass
+import sys
+from io import StringIO
+import time
+from .data import TestResult
+from .decorators import TestInfos, Test
+from rich.console import Console
+from .formater import format_module_result, format_failed
+
+def run_single_test(test_info: TestInfos) -> TestResult:
+  old_stdout = sys.stdout
+  old_stderr = sys.stderr
+  captured_stdout = StringIO()
+  captured_stderr = StringIO()
+  
+  try:
+    sys.stdout = captured_stdout
+    sys.stderr = captured_stderr
+    
+    start_time = time.time()
+    test_info()
+    execution_time = time.time() - start_time
+    
+    return TestResult(
+      name=test_info.name,
+      fail_with=test_info.fail_with,
+      location=test_info.location,
+      failed=test_info.fail_with != None,
+      stdout=captured_stdout.getvalue(),
+      stderr=captured_stderr.getvalue(),
+      execution_time=execution_time,
+      exception=None
+    )
+
+  except Exception as e:
+    execution_time = time.time() - start_time
+    return TestResult(
+      name=test_info.name,
+      fail_with=test_info.fail_with,
+      location=test_info.location,
+      failed=type(e) != test_info.fail_with,
+      stdout=captured_stdout.getvalue(),
+      stderr=captured_stderr.getvalue(),
+      execution_time=execution_time,
+      exception=e
+    )
+  
+  finally:
+    sys.stdout = old_stdout
+    sys.stderr = old_stderr
+
+
+def run_tests(test: Test, verbose):
+  console = Console()
+  for num_module, module in enumerate(test.registered_tests):
+    tests = test.registered_tests[module]
+    results = []
+    failed = False
+    total_time = 0
+    for t in tests:
+      res = run_single_test(t)
+      if not res.failed: failed = True
+      total_time += res.execution_time
+      results.append(res)
+
+    format_module_result(console, num_module+1, len(test.registered_tests), module, not failed, total_time)
+
+    first = True
+    for res in results:
+      if res.failed: format_failed(console, res, verbose)

surimi-0.0.0.dist-info/METADATA

@@ -0,0 +1,14 @@
+Metadata-Version: 2.4
+Name: surimi
+Version: 0.0.0
+Summary: Simple test library fully written in python
+Author-email: Nicolas Beuve <nicolas@beuve.net>
+License-Expression: MIT
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: click
+Requires-Dist: rich
+Dynamic: license-file
+
+![Surimi](docs/surimi.svg)

surimi-0.0.0.dist-info/RECORD

@@ -0,0 +1,14 @@
+surimi/__init__.py,sha256=1z93OJDVn3LaJD2LYz0DqFZ6r8xITvOGUqKKGHrm2XM,134
+surimi/cli.py,sha256=5u9-icy1Yv3p9R25piL6tQHwrH1GmqwReYbUiIHRS1s,503
+surimi/data.py,sha256=Wbvl9no6QA9O45nZM726Iy4Gq3b0Pw0PP_7LXYUXX40,235
+surimi/formater.py,sha256=zz0Ydx4HvYNyIUp18OmF-RH_jCHyi9OxwRvID7ywuRU,1748
+surimi/registration.py,sha256=K6ZvRGHl27JEpFVvkmrpI3ACknju9Kq9YojR_Ng-5EQ,863
+surimi/runner.py,sha256=Hki-IU767bW5N-aZqRTdaGLMpXabdguD6Bcx7Y5pqeA,1935
+surimi/decorators/__init__.py,sha256=7bNWJzyTV_ncMYCfOyN3aK43p4A6EJqTx3OLHTRZ9gQ,85
+surimi/decorators/test.py,sha256=NXG3wH03M9gcxsxA1VrbuqKHPDbG-4LzGRsDmNsq6lo,1175
+surimi-0.0.0.dist-info/licenses/LICENSE,sha256=D4Faj9KdjajDDawPHgitaDXWJHLJMfekogEojpOwHhc,1064
+surimi-0.0.0.dist-info/METADATA,sha256=AaBHexiddoQOQtvaDe7y7oEaSgKEEmr_uQpKt3svGL8,351
+surimi-0.0.0.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+surimi-0.0.0.dist-info/entry_points.txt,sha256=bNaGrLlNoi-OlecfJEN5YiexryK_gzRy6ISHHK4jI-0,43
+surimi-0.0.0.dist-info/top_level.txt,sha256=cXDvYflcNDn4ox5aWNzhAzoUjvvjKxvKGFeMaaa8-NA,7
+surimi-0.0.0.dist-info/RECORD,,

surimi-0.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (84.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

surimi-0.0.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+surimi = surimi.cli:test

surimi-0.0.0.dist-info/licenses/LICENSE

@@ -0,0 +1,7 @@
+Copyright © 2026 Nicolas Beuve
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

surimi-0.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+surimi