Making lint happier and renaming override test function

Author: PrimozGodec

Orange/widgets/visualize/tests/test_owscatterplot.py

@@ -139,17 +139,17 @@ def test_data_column_infs(self):
         attr_x = self.widget.controls.attr_x
         simulate.combobox_activate_item(attr_x, "b")
 
-    def test_regression_line(self):
+    def test_regression_line_pair(self):
         """It is possible to draw the line only for pair of continuous attrs"""
         self.send_signal(self.widget.Inputs.data, self.data)
         self.assertTrue(self.widget.cb_reg_line.isEnabled())
-        self.assertIsNone(self.widget.graph.reg_line_item)
+        self.assertListEqual([], self.widget.graph.reg_line_items)
         self.widget.cb_reg_line.setChecked(True)
-        self.assertIsNotNone(self.widget.graph.reg_line_item)
+        self.assertEqual(4, len(self.widget.graph.reg_line_items))
         self.widget.cb_attr_y.activated.emit(4)
         self.widget.cb_attr_y.setCurrentIndex(4)
         self.assertFalse(self.widget.cb_reg_line.isEnabled())
-        self.assertIsNone(self.widget.graph.reg_line_item)
+        self.assertListEqual([], self.widget.graph.reg_line_items)
 
     def test_points_combo_boxes(self):
         """Check Point box combo models and values"""
@@ -277,7 +277,7 @@ def test_points_selection(self):
         self.assertIsNone(selected_data)
 
     def test_migrate_selection(self):
-        settings = dict(selection=list(range(2)))
+        settings = {"selection": list(range(2))}
         OWScatterPlot.migrate_settings(settings, 0)
         self.assertEqual(settings["selection_group"], [(0, 1), (1, 1)])
 
@@ -371,7 +371,7 @@ def test_vizrank(self):
         vizrank = ScatterPlotVizRank(self.widget)
         n_states = len(data.domain.attributes)
         n_states = n_states * (n_states - 1) / 2
-        states = [state for state in vizrank.iterate_states(None)]
+        states = list(vizrank.iterate_states(None))
         self.assertEqual(len(states), n_states)
         self.assertEqual(len(set(states)), n_states)
         self.assertIsNotNone(vizrank.compute_score(states[0]))
@@ -381,8 +381,7 @@ def test_vizrank(self):
         data = Table("housing")[::10]
         self.send_signal(self.widget.Inputs.data, data)
         vizrank = ScatterPlotVizRank(self.widget)
-        states = [state for state in vizrank.iterate_states(None)]
-        self.assertIsNotNone(vizrank.compute_score(states[0]))
+        self.assertIsNotNone(vizrank.compute_score(next(vizrank.iterate_states(None))))
 
     def test_vizrank_class_nan(self):
         """
@@ -646,29 +645,29 @@ def test_tooltip(self):
                 widget.tooltip_shows_all = False
                 self.assertTrue(graph.help_event(event))
                 (_, text), _ = show_text.call_args
-                self.assertIn("age = {}".format(data[42, "age"]), text)
-                self.assertIn("gender = {}".format(data[42, "gender"]), text)
-                self.assertNotIn("max HR = {}".format(data[42, "max HR"]), text)
+                self.assertIn(f"age = {data[42, 'age']}", text)
+                self.assertIn(f"gender = {data[42, 'gender']}", text)
+                self.assertNotIn(f"max HR = {data[42, 'max HR']}", text)
                 self.assertNotIn("others", text)
 
                 # Show all attributes
                 widget.tooltip_shows_all = True
                 self.assertTrue(graph.help_event(event))
                 (_, text), _ = show_text.call_args
-                self.assertIn("age = {}".format(data[42, "age"]), text)
-                self.assertIn("gender = {}".format(data[42, "gender"]), text)
-                self.assertIn("max HR = {}".format(data[42, "max HR"]), text)
+                self.assertIn(f"age = {data[42, 'age']}", text)
+                self.assertIn(f"gender = {data[42, 'gender']}", text)
+                self.assertIn(f"max HR = {data[42, 'max HR']}", text)
                 self.assertIn("... and 4 others", text)
 
             # Two points hovered
             with patch.object(scatterplot_item, "pointsAt",
                               return_value=[all_points[42], all_points[100]]):
                 self.assertTrue(graph.help_event(event))
                 (_, text), _ = show_text.call_args
-                self.assertIn("age = {}".format(data[42, "age"]), text)
-                self.assertIn("gender = {}".format(data[42, "gender"]), text)
-                self.assertIn("age = {}".format(data[100, "age"]), text)
-                self.assertIn("gender = {}".format(data[100, "gender"]), text)
+                self.assertIn(f"age = {data[42, 'age']}", text)
+                self.assertIn(f"gender = {data[42, 'gender']}", text)
+                self.assertIn(f"age = {data[100, 'age']}", text)
+                self.assertIn(f"gender = {data[100, 'gender']}", text)
 
             # No points hovered
             with patch.object(scatterplot_item, "pointsAt",
@@ -694,10 +693,10 @@ def prepare_data():
                 data.Y = np.array(values * 10, dtype=float)
             return data
 
-        def assert_equal(data, max):
+        def assert_equal(data, max_):
             self.send_signal(self.widget.Inputs.data, data)
-            pen_data, brush_data = self.widget.graph.get_colors()
-            self.assertEqual(max, len(np.unique([id(p) for p in pen_data])), )
+            pen_data, _ = self.widget.graph.get_colors()
+            self.assertEqual(max_, len(np.unique([id(p) for p in pen_data])), )
 
         assert_equal(prepare_data(), MAX_COLORS)
         # data with nan value