format

tests/datetimeex/test_duration_yaml.py

@@ -4,39 +4,46 @@ import pytest
 from ioex.datetimeex import Duration
 yaml = pytest.importorskip('yaml')
 
+
 @pytest.mark.parametrize(('loader'), [yaml.Loader, yaml.SafeLoader])
 @pytest.mark.parametrize(('expected_duration', 'yaml_string'), [
-    [Duration(years = 0),  '!duration\nyears: 0'],
-    [Duration(years = 32), '!duration\nyears: 32'],
-    [Duration(years = 0),  '!duration\n{}'],
-    [Duration(years = 0),  '!duration {}'],
-    ])
+    [Duration(years=0),  '!duration\nyears: 0'],
+    [Duration(years=32), '!duration\nyears: 32'],
+    [Duration(years=0),  '!duration\n{}'],
+    [Duration(years=0),  '!duration {}'],
+])
 def test_from_yaml(expected_duration, yaml_string, loader):
     class TestLoader(loader):
         pass
     Duration.register_yaml_constructor(TestLoader)
-    loaded_duration = yaml.load(yaml_string, Loader = TestLoader)
+    loaded_duration = yaml.load(yaml_string, Loader=TestLoader)
     assert expected_duration == loaded_duration
 
+
 @pytest.mark.parametrize(('loader'), [yaml.Loader, yaml.SafeLoader])
 @pytest.mark.parametrize(('expected_duration', 'yaml_string', 'tag'), [
-    [Duration(years = 2),  '!dur\nyears: 2', '!dur'],
-    [Duration(years = 0), '!duration_tag {}', '!duration_tag'],
-    ])
+    [Duration(years=2),  '!dur\nyears: 2', '!dur'],
+    [Duration(years=0), '!duration_tag {}', '!duration_tag'],
+])
 def test_from_yaml_tag(expected_duration, yaml_string, tag, loader):
     class TestLoader(loader):
         pass
-    Duration.register_yaml_constructor(TestLoader, tag = tag)
-    loaded_duration = yaml.load(yaml_string, Loader = TestLoader)
+    Duration.register_yaml_constructor(TestLoader, tag=tag)
+    loaded_duration = yaml.load(yaml_string, Loader=TestLoader)
     assert expected_duration == loaded_duration
 
+
 @pytest.mark.parametrize(('dumper'), [yaml.Dumper, yaml.SafeDumper])
 @pytest.mark.parametrize(('duration', 'yaml_string'), [
-    [Duration(years = 0),  '!duration {}\n'],
-    [Duration(years = 32), '!duration\nyears: 32\n'],
-    ])
+    [Duration(years=0),  '!duration {}\n'],
+    [Duration(years=32), '!duration\nyears: 32\n'],
+])
 def test_to_yaml(duration, yaml_string, dumper):
     class TestDumper(dumper):
         pass
     Duration.register_yaml_representer(TestDumper)
-    assert yaml.dump(duration, Dumper = TestDumper, default_flow_style = False) == yaml_string
+    assert yaml_string == yaml.dump(
+        duration,
+        Dumper=TestDumper,
+        default_flow_style=False,
+    )