def _test(self, row, yp, yv): reals = np.arange(6) + 100 grid = ENDGame(None, row, reals, None) result_yp, result_yv = grid._y_vectors() self.assertArrayEqual(result_yp, yp) self.assertArrayEqual(result_yv, yv)
def test_class_attributes(self): reals = np.arange(6) + 100 grid = ENDGame(None, None, reals, None) self.assertEqual(grid._v_offset, -0.5)