From ea81714867350f3884055bf9dba8e386c58fb81b Mon Sep 17 00:00:00 2001 From: Zekun Shi Date: Fri, 27 Oct 2023 16:36:25 +0800 Subject: [PATCH] fix lint --- tests/integral/pw/pw_test.py | 4 +++- tests/native/xla/example_test.py | 1 + tests/system/crystal_test.py | 8 ++++++-- 3 files changed, 10 insertions(+), 3 deletions(-) diff --git a/tests/integral/pw/pw_test.py b/tests/integral/pw/pw_test.py index 0a43926..31db8c3 100644 --- a/tests/integral/pw/pw_test.py +++ b/tests/integral/pw/pw_test.py @@ -25,7 +25,9 @@ class PWTest(parameterized.TestCase): - @parameterized.parameters(("C2", 3.5667, 76.5484253352856),) + @parameterized.parameters( + ("C2", 3.5667, 76.5484253352856), + ) def test_init_from_name_and_lattice( self, crystal_name: str, a: float, expected_vol: float ): diff --git a/tests/native/xla/example_test.py b/tests/native/xla/example_test.py index 123c545..f211026 100644 --- a/tests/native/xla/example_test.py +++ b/tests/native/xla/example_test.py @@ -7,6 +7,7 @@ import numpy as np from absl import logging from absl.testing import absltest + from d4ft.native.xla.custom_call import CustomCallMeta from tests.native.xla.example import Parent, _Example, _ExampleMember diff --git a/tests/system/crystal_test.py b/tests/system/crystal_test.py index 359a5f5..1d9ed00 100644 --- a/tests/system/crystal_test.py +++ b/tests/system/crystal_test.py @@ -21,7 +21,9 @@ class CrystalTest(parameterized.TestCase): - @parameterized.parameters(("C2", 3.5667, 76.5484253352856),) + @parameterized.parameters( + ("C2", 3.5667, 76.5484253352856), + ) def test_init_from_name_and_lattice( self, crystal_name: str, a: float, expected_vol: float ): @@ -39,7 +41,9 @@ def test_init_from_name_and_lattice( self.assertAlmostEqual(crystal.vol, expected_vol) - @parameterized.parameters(("diamond.xyz", 76.5484253352856),) + @parameterized.parameters( + ("diamond.xyz", 76.5484253352856), + ) def test_init_from_xyz_file(self, xyz_file: str, expected_vol: float): crystal = Crystal.from_xyz_file(xyz_file) logging.info(f'cell volume:{crystal.vol}')