rbx.cp 0.5.2__py3-none-any.whl → 0.5.4__py3-none-any.whl

rbx/box/environment.py

@@ -212,6 +212,7 @@ def install_environment(name: str, file: pathlib.Path):
         )
         raise typer.Exit(1)
 
+    get_environment_path(name).parent.mkdir(parents=True, exist_ok=True)
     get_environment_path(name).write_bytes(file.read_bytes())
     console.console.print(
         f'[success]Environment [item]{name}[/item] was installed from [item]{file}[/item].'

rbx/box/presets/__init__.py

@@ -316,6 +316,7 @@ def _install(root: pathlib.Path = pathlib.Path(), force: bool = False):
                 should_copy_env = False
 
         if should_copy_env:
+            get_environment_path(preset.name).parent.mkdir(parents=True, exist_ok=True)
             shutil.rmtree(get_environment_path(preset.name), ignore_errors=True)
             shutil.copyfile(str(root / preset.env), get_environment_path(preset.name))
 

rbx/box/stresses.py

@@ -1,4 +1,3 @@
-import dataclasses
 import functools
 import time
 from shutil import rmtree
@@ -175,11 +174,13 @@ def run_stress(
 
         @functools.cache
         def run_solution_and_checker_fn(
-            solution: str,
-            checker: Optional[finder_parser.FinderChecker],
+            call: finder_parser.FinderCall,
             input_path=input_path,
             expected_output_path=expected_output_path,
         ) -> finder_parser.FinderResult:
+            solution = call.solution
+            checker = call.checker
+
             solution_result = run_solution_fn(solution)
 
             if checker is None:
@@ -196,19 +197,11 @@ def run_stress(
                 solution=solution,
                 outcome=checker_result.outcome,
                 checker=checker,
-                truth_value=True,
                 solution_result=solution_result,
                 checker_result=checker_result,
             )
 
-        def run_fn(
-            call: finder_parser.FinderCall,
-        ) -> finder_parser.FinderResult:
-            finder_result = run_solution_and_checker_fn(call.solution, call.checker)
-            truth_value = call.expected_outcome.match(finder_result.outcome)
-            return dataclasses.replace(finder_result, truth_value=truth_value)
-
-        runner = finder_parser.FinderTreeRunner(runner=run_fn)
+        runner = finder_parser.FinderTreeRunner(runner=run_solution_and_checker_fn)
         finder_outcome: finder_parser.FinderOutcome = runner.transform(parsed_finder)
 
         internal_error_results = [

rbx/box/stressing/finder_parser.py

@@ -2,7 +2,7 @@ import dataclasses
 import pathlib
 import typing
 from enum import Enum
-from typing import Callable, List, Optional
+from typing import Callable, List, Optional, Union
 
 import lark
 import typer
@@ -20,23 +20,34 @@ disjunction: conjunction | disjunction _OR conjunction
 
 conjunction: _atom | conjunction _AND _atom
 
-_atom: statement | "(" disjunction ")" | negation
+_atom: logical | "(" disjunction ")" | negation
 negation: _NOT "(" disjunction ")"
 
-statement: solution matcher outcome checking?
+// Expressions
+logical: eval matcher expected_outcome -> matching
+       | eval equality (eval | outcome) -> equating
 
+eval: "[" solution checking? "]"
+
+// Eval
 solution: _filename | WILDCARD
-outcome: CNAME
 checking: "ON"i (checking_mode? checker | ":nil")
 checking_mode: MODE ":"
 MODE: "2" | "3"
 checker: _filename | WILDCARD
 
+// Outcomes
+expected_outcome: CNAME
+outcome: CNAME
+
 // Operators
 matcher: MATCHES | NOT_MATCHES
+equality: EQUALS | NOT_EQUALS
 
 MATCHES: "~"
 NOT_MATCHES: "!~"
+EQUALS: "=="
+NOT_EQUALS: "!="
 _OR: "||"
 _AND: "&&"
 _NOT: "!"
@@ -46,7 +57,14 @@ WILDCARD: "$"
 _filename: FILENAME | "\"" FILENAME "\""
 FILENAME: /[\/A-Za-z0-9\-_\.]/+
 
-%import common.CNAME
+// Names (Variables)
+LCASE_LETTER: "a".."z"
+UCASE_LETTER: "A".."Z"
+DIGIT: "0".."9"
+LETTER: UCASE_LETTER | LCASE_LETTER
+WORD: LETTER+
+CNAME: ("_"|LETTER) ("_"|LETTER|DIGIT|"+")*
+
 %ignore " "
 """
 
@@ -67,7 +85,6 @@ class FinderChecker:
 @dataclasses.dataclass(frozen=True)
 class FinderCall:
     solution: str
-    expected_outcome: ExpectedOutcome
     checker: Optional[FinderChecker]
 
 
@@ -83,7 +100,6 @@ class FinderResult:
     solution: str
     outcome: Outcome
     checker: Optional[FinderChecker]
-    truth_value: bool
 
     # Auxiliary information.
     solution_result: Optional[FinderSolutionResult] = None
@@ -155,8 +171,8 @@ def _get_checker_from_token(token: lark.Token) -> str:
     return path
 
 
-def _get_statement_checker(statement: lark.ParseTree) -> Optional[FinderChecker]:
-    checking_nodes = list(statement.find_data('checking'))
+def _get_eval_checker(eval: lark.ParseTree) -> Optional[FinderChecker]:
+    checking_nodes = list(eval.find_data('checking'))
     if not checking_nodes:
         return _get_default_checker_for_finder()
     (checking,) = checking_nodes
@@ -227,11 +243,11 @@ def get_all_solution_items(tree: lark.ParseTree) -> List[CodeItem]:
 
 
 def _get_all_finder_checkers(tree: lark.ParseTree) -> List[FinderChecker]:
-    statement_nodes = tree.find_data('statement')
+    eval_nodes = tree.find_data('eval')
     res = []
 
-    for statement_node in statement_nodes:
-        finder_checker = _get_statement_checker(statement_node)
+    for eval_node in eval_nodes:
+        finder_checker = _get_eval_checker(eval_node)
         if finder_checker is not None:
             res.append(finder_checker)
 
@@ -299,8 +315,6 @@ def validate(tree: lark.ParseTree):
 
 @lark.v_args(inline=True)
 class FinderTreeRunner(lark.Transformer):
-    outcome = ExpectedOutcome
-
     def __init__(
         self,
         runner: Callable[[FinderCall], FinderResult],
@@ -310,27 +324,72 @@ class FinderTreeRunner(lark.Transformer):
     def solution(self, token: lark.Token) -> str:
         return _get_solution_from_token(token)
 
+    def outcome(self, token: lark.Token) -> Outcome:
+        try:
+            outcome = Outcome(token.value)
+        except ValueError:
+            try:
+                expected_outcome = self.expected_outcome(token)
+            except ValueError:
+                raise ValueError(f'"{token.value}" is not a valid Outcome.') from None
+            outcomes = expected_outcome.get_matches()
+            if len(outcomes) != 1:
+                raise ValueError(
+                    f'"{token.value}" is not a valid Outcome. You are trying to specify an ExpectedOutcome, instead of a single Outcome.'
+                ) from None
+            return outcomes[0]
+        return outcome
+
+    def expected_outcome(self, token: lark.Token) -> ExpectedOutcome:
+        return ExpectedOutcome(token.value)
+
     def matcher(self, op: lark.Token) -> bool:
         return op.value == '~'
 
+    def equality(self, op: lark.Token) -> bool:
+        return op.value == '=='
+
     @lark.v_args(inline=False, tree=True)
-    def statement(
+    def eval(self, tree: lark.ParseTree) -> FinderResult:
+        solution = typing.cast(str, tree.children[0])
+        checker: Optional[FinderChecker] = _get_eval_checker(tree)
+
+        call = FinderCall(solution, checker=checker)
+        return self.run_fn(call)
+
+    def matching(
         self,
-        tree: lark.ParseTree,
+        eval_result: FinderResult,
+        is_positive: bool,
+        expected_outcome: ExpectedOutcome,
     ) -> FinderOutcome:
-        solution = typing.cast(str, tree.children[0])
-        is_positive = typing.cast(bool, tree.children[1])
-        expected_outcome = typing.cast(ExpectedOutcome, tree.children[2])
+        truth_value = expected_outcome.match(eval_result.outcome)
+        if not is_positive:
+            truth_value = not truth_value
+
+        return FinderOutcome(truth_value=truth_value, results=[eval_result])
+
+    def equating(
+        self,
+        eval_result: FinderResult,
+        is_positive: bool,
+        result_or_outcome: Union[FinderResult, Outcome],
+    ) -> FinderOutcome:
+        results = [eval_result]
+        truth_value = True
 
-        checker: Optional[FinderChecker] = _get_statement_checker(tree)
+        if isinstance(result_or_outcome, Outcome):
+            outcome: Outcome = result_or_outcome
+            truth_value = eval_result.outcome == outcome
+        else:
+            result: FinderResult = result_or_outcome
+            truth_value = eval_result.outcome == result.outcome
+            results.append(result)
 
-        call = FinderCall(solution, expected_outcome=expected_outcome, checker=checker)
-        result = self.run_fn(call)
-        truth_value = result.truth_value
         if not is_positive:
             truth_value = not truth_value
 
-        return FinderOutcome(truth_value=truth_value, results=[result])
+        return FinderOutcome(truth_value=truth_value, results=results)
 
     def negation(self, value: FinderOutcome) -> FinderOutcome:
         return dataclasses.replace(value, truth_value=not value.truth_value)

rbx/resources/presets/default/problem/problem.rbx.yml

@@ -38,7 +38,7 @@ stresses:
     generator:
       name: 'gen'
       args: '[1..<MAX_N>] @' # `@` generates a random string
-    finder: 'sols/wa.cpp ~ INCORRECT'
+    finder: '[sols/wa.cpp] ~ INCORRECT'
 vars:
   "MAX_N": 1000000000  # Can be used in the validator, in stress tests and in the statement.
 

{rbx_cp-0.5.2.dist-info → rbx_cp-0.5.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rbx.cp
-Version: 0.5.2
+Version: 0.5.4
 Summary: 
 Author: Roberto Sales
 Requires-Python: >=3.9,<4.0

{rbx_cp-0.5.2.dist-info → rbx_cp-0.5.4.dist-info}/RECORD

@@ -17,7 +17,7 @@ rbx/box/contest/schema.py,sha256=rxzjJasMWPKKhvSJs4eW4A2oCiA4gXgfF-MzqsbPslQ,491
 rbx/box/contest/statements.py,sha256=iG_E4bqhc7hof4LlAliw4m9VO8HcDzvnhsWmxzVRQM4,2690
 rbx/box/creation.py,sha256=mZ7kEt1F-0scwDBc6oWsJxFRW69AccZlKuJfN1QyDd4,2203
 rbx/box/download.py,sha256=MFP-R26JiYGAP89I0TK-0fYc69Fsd20tsBqgtRCy5AE,2234
-rbx/box/environment.py,sha256=HiufDkaXJeZk9gWvt27UMmyzaknnODu8flGerrwnNhE,10851
+rbx/box/environment.py,sha256=i-IGHDyFDaYF1fyr0QHdGZ3z4NWp5sj6nxsbnxhAYq4,10924
 rbx/box/extensions.py,sha256=p0iLaU28KswOBDX2HGVO_dR2gk-JSAWb6sXC6GZ1d0w,738
 rbx/box/generators.py,sha256=UJNyvwbY4JBVpmdMoVsYi16O3oodMgguH_dmZAoOFm4,16104
 rbx/box/generators_test.py,sha256=mQqHepAMYa6zV_PseQALI0nIX6AdQktt6lh94muFhNw,1758
@@ -31,7 +31,7 @@ rbx/box/packaging/packager.py,sha256=suCT_SLnWa915rV2j8VFqzH43HGKRTr9mGGlrvj45aw
 rbx/box/packaging/polygon/packager.py,sha256=CcjHzDr4MwSyp270gsPY6RWoz8bUJeykDqXPvQ3XZ1U,10773
 rbx/box/packaging/polygon/test.py,sha256=bgEju5PwudgyfwxXJagm8fM6CJVlWM6l_-2q1V-oKaQ,3069
 rbx/box/packaging/polygon/xml_schema.py,sha256=-r24bCeRMGLrGGoT9FIgmqr87xHL-JzrFaR6bztbYtw,2703
-rbx/box/presets/__init__.py,sha256=kV_XJI2PQV6ShT07pr-gCvdKDm-8BZ6SsOtXO9dIY5A,16485
+rbx/box/presets/__init__.py,sha256=bDmaBWnwh71lkn2pfGIREN-xlo3HpMZBN9ucodbxscc,16573
 rbx/box/presets/fetch.py,sha256=F-BCOlvEBEyDqtOhiDuGPn4EDtA4Bwm-fqHJ7zZGlW8,1975
 rbx/box/presets/lock_schema.py,sha256=6sRPnyePOC8yy-5WcD5JRZdDJHf8loqbvpQ1IPiOU9s,349
 rbx/box/presets/schema.py,sha256=mZmSPkQsw7eQM0lQN6er1MO_LiW1ObwwAZFDK0F5fxE,1962
@@ -45,9 +45,9 @@ rbx/box/statements/joiners.py,sha256=ZbxomnMjEFT8yf5WSWUB4tBa3DL3AhjGEuh8uqHyDdg
 rbx/box/statements/latex.py,sha256=LkcHwXjMFxbw--Gj9T1VkFKQFsXhY9dN7xZHpZycNW8,1346
 rbx/box/statements/latex_jinja.py,sha256=7WBfn1h8DpqCAmSE6Av64HfURMnJ2AO4QX1CD72sz5E,7096
 rbx/box/statements/schema.py,sha256=g3KgBn4nIqx-0utH8R2FCqPmJP969chhYfn96chQgd4,3851
-rbx/box/stresses.py,sha256=9kkGw6DAUo6xXa98MT6ww6H5IIM0uPQtE8AWCrvIRto,10945
+rbx/box/stresses.py,sha256=mbKa70QyPOmgXfhGcArTKsdXjH0PFDVmydY3W8bgZlk,10596
 rbx/box/stressing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rbx/box/stressing/finder_parser.py,sha256=KVUasnqtiUc5TA5uuO-pAwXyIOflkWuSUOR1Pyn_Ans,10117
+rbx/box/stressing/finder_parser.py,sha256=Hw9m2viRGnVqVgO39D51s2qAEjZK1LKWe94rkqCx7VA,11919
 rbx/box/stressing/generator_parser.py,sha256=oHZryjR3YohgaSO9WEirQ7b2e-98WgZStF0N99W4Thw,7380
 rbx/box/testcases.py,sha256=bi7T5xXkwyWOkoI6ILcaf2gSSVuuNtZjhP5yL0DJAu4,1452
 rbx/box/ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -116,7 +116,7 @@ rbx/resources/presets/default/contest/statement/template.rbx.tex,sha256=XSGtd8JJ
 rbx/resources/presets/default/preset.rbx.yml,sha256=hYesK08n17FBNThr4iaGrDfC0L3us_pzGDkhD33Ryt4,313
 rbx/resources/presets/default/problem/.gitignore,sha256=zc-lnGQQZsLBaXpSshesA_QfxhiZdNCSJDEuKqlPtjw,29
 rbx/resources/presets/default/problem/gen.cpp,sha256=rn6sGRjZ1sFE1Rq02r6488iquY9xTrutcvLv4d1sohA,178
-rbx/resources/presets/default/problem/problem.rbx.yml,sha256=h423WK90gj69OVSXQQOx0AaBjAjXtW6yOZZg5BPpiB0,1377
+rbx/resources/presets/default/problem/problem.rbx.yml,sha256=sgOTa7LedlY3nFOno28XHAgUN89w9p5dl0aIafq1dQs,1379
 rbx/resources/presets/default/problem/random.py,sha256=-iPorU24QHfp39EYRJX9jMKcTIxxz5ejKoAzPLIuu1g,98
 rbx/resources/presets/default/problem/random.txt,sha256=oDS_a9LdKSTKj4MvGLWvHSljI3AjRY6MclaxEwK794I,23
 rbx/resources/presets/default/problem/sols/main.cpp,sha256=AW-j65DiFYUN18rddTKCWc_VyYCMgCbjZ0jAJ-0JLuA,124
@@ -157,8 +157,8 @@ rbx/testdata/caching/executable.py,sha256=WKRHNf_fprFJd1Fq1ubmQtR3mZzTYVNwKPLWuZ
 rbx/testdata/compatible,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rbx/testing_utils.py,sha256=ZZLKMUHlZ4HwsuNY50jqSBJ9HhpnFdba7opjDsvXE1U,2084
 rbx/utils.py,sha256=OkOLjNG8Pq7y2ylqhXQyAOVDfUsva_4Mfjd8OmII55E,4145
-rbx_cp-0.5.2.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-rbx_cp-0.5.2.dist-info/METADATA,sha256=fiqGkYBfzJBTozxW_9jKubvmcVyVioblqa4iREtVUVI,3253
-rbx_cp-0.5.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-rbx_cp-0.5.2.dist-info/entry_points.txt,sha256=qBTLBOeifT1F00LWaEewRRE_jQPgvH7BUdJfZ-dYsFU,57
-rbx_cp-0.5.2.dist-info/RECORD,,
+rbx_cp-0.5.4.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+rbx_cp-0.5.4.dist-info/METADATA,sha256=M7IsN4Rby6iiSRtncAAu_JTiiw-Yi_tNXFYohcCDHhA,3253
+rbx_cp-0.5.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+rbx_cp-0.5.4.dist-info/entry_points.txt,sha256=qBTLBOeifT1F00LWaEewRRE_jQPgvH7BUdJfZ-dYsFU,57
+rbx_cp-0.5.4.dist-info/RECORD,,