From 5e563d82090733279f74629ff70583ca1e3039fa Mon Sep 17 00:00:00 2001 From: samwaseda <o.waseda@mpie.de> Date: Thu, 30 Jan 2025 17:36:12 +0000 Subject: [PATCH] run black --- tests/unit/test_parser.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/tests/unit/test_parser.py b/tests/unit/test_parser.py index 7cf1fa7..a9b1469 100644 --- a/tests/unit/test_parser.py +++ b/tests/unit/test_parser.py @@ -229,15 +229,17 @@ class Input: T: u(float, units="kelvin") n: int + @semantikon_class @dataclass class Output: E: u(float, units="electron_volt") L: u(float, units="angstrom") + @Workflow.wrap.as_function_node def run_md(inp: Input) -> Output: - out = Output(E=1., L=2.) + out = Output(E=1.0, L=2.0) # This line should be removed with the next version of semantikon out._is_semantikon_class = True return out @@ -246,7 +248,7 @@ def run_md(inp: Input) -> Output: class TestDataclass(unittest.TestCase): def test_dataclass(self): wf = Workflow("my_wf") - inp = Input(T=300., n=100) + inp = Input(T=300.0, n=100) # This line should be removed with the next version of semantikon inp._is_semantikon_class = True wf.node = run_md(inp) @@ -255,12 +257,8 @@ def test_dataclass(self): i_txt = "my_wf.node.inputs.inp" o_txt = "my_wf.node.outputs.out" triples = ( - ( - URIRef(f'{i_txt}.n.value'), - RDFS.subClassOf, - URIRef(f'{i_txt}.value') - ), - (URIRef(o_txt), PNS.hasValue, URIRef(f'{o_txt}.E.value')) + (URIRef(f"{i_txt}.n.value"), RDFS.subClassOf, URIRef(f"{i_txt}.value")), + (URIRef(o_txt), PNS.hasValue, URIRef(f"{o_txt}.E.value")), ) for triple in triples: self.assertEqual( @@ -268,6 +266,5 @@ def test_dataclass(self): ) - if __name__ == "__main__": unittest.main()