Merge pull request #156 from alexanderthclark/codex/refactor-tests-for-improved-code-quality

Remove __main__ blocks and tidy EOF newlines

Author: alexanderthclark

tests/__init__.py

@@ -0,0 +1 @@
+

tests/test_auction.py

@@ -103,5 +103,3 @@ def test_clearing_price_correctness(self):
         auction = DoubleAuction(*agents)
         self.assertEqual(auction.q, 2)
         self.assertEqual(auction.p, (6, 9))
-
-

tests/test_curves.py

@@ -244,4 +244,3 @@ def test_numeric_coefficients(self):
         for q in [-1, 0, 1, 2]:
             expected = 1 + 2 * q + 3 * q**2
             self.assertAlmostEqual(curve(q), expected)
-

tests/test_domain_conventions.py

@@ -82,7 +82,3 @@ def test_affine_element_domain(self):
         self.assertTrue(elem.in_domain(0))
         self.assertTrue(elem.in_domain(10))
         self.assertFalse(elem.in_domain(11))
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/test_games.py

@@ -247,7 +247,3 @@ def test_class_helpers(self):
             rps.action_names,
             (("Rock", "Paper", "Scissors"), ("Rock", "Paper", "Scissors")),
         )
-
-
-if __name__ == '__main__':
-    unittest.main()

tests/test_horizontal_curves.py

@@ -96,7 +96,3 @@ def test_equilibrium_blocks_horizontal_demand(self):
 
             self.assertIn("perfectly elastic", str(cm.exception))
             self.assertIn("demand", str(cm.exception).lower())
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/test_intersection_operator.py

@@ -59,7 +59,3 @@ def test_with_formula_curves(self):
         self.assertIsInstance(market, Market)
         self.assertAlmostEqual(market.p, 10.0, places=2)
         self.assertAlmostEqual(market.q, 5.0, places=2)
-
-
-if __name__ == '__main__':
-    unittest.main()

tests/test_longrun.py

@@ -17,7 +17,3 @@ def test_basic_attributes(self):
         self.assertAlmostEqual(lr.firm_q, 1.0)
         self.assertAlmostEqual(lr.market_q, 8.0)
         self.assertAlmostEqual(lr.n_firms, 8.0)
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/test_monopoly.py

@@ -156,7 +156,3 @@ def test_kinked_demand_both_segments(self):
         # Verify we found the optimum
         self.assertGreater(m.profit, 0.99 * max_profit_grid)
         self.assertAlmostEqual(m.q, best_q_grid, delta=0.1)
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/test_plotting.py

@@ -73,4 +73,3 @@ def test_update_axes_limits_uses_current_axes(self):
         result = update_axes_limits()
 
         self.assertIs(result, ax)
-