formatting

tests/test_assets.py

@@ -2,6 +2,8 @@ from grum import assets
 
 
 def test_get_asset():
-    
+    ret = assets.get_asset("testname")
-    ret = assets.get_asset('testname')
+    assert ret.endswith("grum/grum/assets/testname")
-    assert ret.endswith('grum/grum/assets/testname')
+
+
+

tests/test_plotdesc.py

@@ -1,77 +1,84 @@
-from grum.plotdesc import PlotDescription
-import pytest
-import pyqtgraph as pg
-from grum.theme import pg_plot_style
-from grum.mdi.mdisubplot import MDISubPlot
-from grum.mainwin import MainWindow
-
 import sys
+import pytest
 from PyQt5.QtWidgets import QApplication
+import pyqtgraph as pg
+
 from grum import theme
+from grum.mainwin import MainWindow
+from grum.mdi.mdisubplot import MDISubPlot
+from grum.plotdesc import PlotDescription
+from grum.theme import pg_plot_style
 
 
 @pytest.mark.parametrize(
-    "plot_name, plot_title, plot_xlabel, plot_ylabel, xs,ys", [('plot_name', 
+    "plot_name, plot_title, plot_xlabel, plot_ylabel, xs,ys",
-    'plot_title', 
+    [
-    'plot_xlabel', 
+        ("plot_name", "plot_title", "plot_xlabel", "plot_ylabel", [1, 2], [3, 4]),
-    'plot_ylabel', 
+        ("plot_name", None, None, None, [], []),
-    [1, 2], 
+    ],
-    [3, 4]), 
-    ('plot_name', 
-    None, 
-    None, 
-    None, 
-    [], 
-    [])]
 )
-def test_plot_derscription_init(plot_name, plot_title, plot_xlabel, plot_ylabel, xs, ys):
+def test_plot_derscription_init(
-
+    plot_name, plot_title, plot_xlabel, plot_ylabel, xs, ys
+):
     if plot_title:
-        pd = PlotDescription(plot_name, plot_title,
+        pd = PlotDescription(plot_name, plot_title, plot_xlabel, plot_ylabel, xs, ys)
-                         plot_xlabel, plot_ylabel, xs, ys)
+    else:
-    else: 
         pd = PlotDescription(plot_name)
-    
+
     assert pd.name == plot_name
     assert pd.title == plot_title
     assert pd.xlabel == plot_xlabel
     assert pd.ylabel == plot_ylabel
     assert pd.xs == xs
     assert pd.ys == ys
-   
-def test_data():
-    pd = PlotDescription('name', xs = [1,2], ys = [3,4])
 
-    assert pd.data == ([1,2], [3,4])
+
+def test_data():
+    pd = PlotDescription("name", xs=[1, 2], ys=[3, 4])
+
+    assert pd.data == ([1, 2], [3, 4])
+
 
 def test_append():
-    pd = PlotDescription('name', xs = [1,2], ys = [3,4])
+    pd = PlotDescription("name", xs=[1, 2], ys=[3, 4])
+    pd.append([3, 5])
 
-    pd.append([3,5])    
+    assert pd.xs == [1, 2, 3]
+    assert pd.ys == [3, 4, 5]
 
-    assert pd.xs == [1,2,3]
-    assert pd.ys == [3,4,5]
 
 def test_make_plot():
-
+    pd = PlotDescription(
-    pd = PlotDescription('plot_name', 'plot_title', 'plot_xlabel', 'plot_ylabel', [1,2], [3,4])
+        "plot_name", "plot_title", "plot_xlabel", "plot_ylabel", [1, 2], [3, 4]
+    )
     app = QApplication(sys.argv)
     theme.apply(app)
     mw = MainWindow(add_examples=True)
 
-    mdi_sub = MDISubPlot('mdi', pd)
+    mdi_sub = MDISubPlot("mdi", pd)
-    pw = pg.PlotWidget() 
+    pw = pg.PlotWidget()
     style = pg_plot_style()
 
     ret = pd.make_plot(pw, style)
-    
+
     assert isinstance(ret, pg.PlotDataItem)
-    assert pw.getAxis("left").labelText == 'plot_ylabel'
+    assert pw.getAxis("left").labelText == "plot_ylabel"
-    assert pw.getAxis("bottom").labelText == 'plot_xlabel'
+    assert pw.getAxis("bottom").labelText == "plot_xlabel"
-    assert pw.getPlotItem().titleLabel.text == 'plot_title'
+    assert pw.getPlotItem().titleLabel.text == "plot_title"
 
 
 def test_to_dict():
-    pd = PlotDescription('plot_name', 'plot_title', 'plot_xlabel', 'plot_ylabel', [1,2], [3,4])
+    pd = PlotDescription(
+        "plot_name", "plot_title", "plot_xlabel", "plot_ylabel", [1, 2], [3, 4]
+    )
+
+    assert pd.to_dict() == {
+        "title": "plot_title",
+        "xlabel": "plot_xlabel",
+        "xs": [1, 2],
+        "ylabel": "plot_ylabel",
+        "ys": [3, 4],
+    }
+
+
 
-    assert pd.to_dict() == {'title': 'plot_title', 'xlabel': 'plot_xlabel', 'xs': [1, 2], 'ylabel': 'plot_ylabel', 'ys': [3,4]}