# TODO(Jerry): Fill out test_nf_glpk() and add a performance test.

import unittest

import swiglpk as glp

from wholecell.utils._netflow.nf_glpk import (
    MessageLevel,
    NetworkFlowGLPK,
    SimplexMethod,
)

# Silence Sphinx autodoc warning
unittest.TestCase.__module__ = "unittest"


class Test_NetworkFlowGLPK(unittest.TestCase):
    def test_swiglpk(self):
        """Test the underlying GLPK lib and its SWIG interface based on
        the example from https://github.com/biosustain/swiglpk
        """
        ia = glp.intArray(1 + 1000)
        ja = glp.intArray(1 + 1000)
        ar = glp.doubleArray(1 + 1000)

        lp = glp.glp_create_prob()
        smcp = glp.glp_smcp()
        glp.glp_init_smcp(smcp)
        smcp.msg_lev = glp.GLP_MSG_ALL  # use GLP_MSG_ERR?

        glp.glp_set_prob_name(lp, "sample")
        glp.glp_set_obj_dir(lp, glp.GLP_MAX)

        glp.glp_add_rows(lp, 3)
        glp.glp_set_row_name(lp, 1, "p")
        glp.glp_set_row_bnds(lp, 1, glp.GLP_UP, 0.0, 100.0)
        glp.glp_set_row_name(lp, 2, "q")
        glp.glp_set_row_bnds(lp, 2, glp.GLP_UP, 0.0, 600.0)
        glp.glp_set_row_name(lp, 3, "r")
        glp.glp_set_row_bnds(lp, 3, glp.GLP_UP, 0.0, 300.0)
        glp.glp_add_cols(lp, 3)
        glp.glp_set_col_name(lp, 1, "x1")
        glp.glp_set_col_bnds(lp, 1, glp.GLP_LO, 0.0, 0.0)
        glp.glp_set_obj_coef(lp, 1, 10.0)
        glp.glp_set_col_name(lp, 2, "x2")
        glp.glp_set_col_bnds(lp, 2, glp.GLP_LO, 0.0, 0.0)
        glp.glp_set_obj_coef(lp, 2, 6.0)
        glp.glp_set_col_name(lp, 3, "x3")
        glp.glp_set_col_bnds(lp, 3, glp.GLP_LO, 0.0, 0.0)
        glp.glp_set_obj_coef(lp, 3, 4.0)

        ia[1] = 1
        ja[1] = 1
        ar[1] = 1.0  # a[1,1] = 1
        ia[2] = 1
        ja[2] = 2
        ar[2] = 1.0  # a[1,2] = 1
        ia[3] = 1
        ja[3] = 3
        ar[3] = 1.0  # a[1,3] = 1
        ia[4] = 2
        ja[4] = 1
        ar[4] = 10.0  # a[2,1] = 10
        ia[5] = 3
        ja[5] = 1
        ar[5] = 2.0  # a[3,1] = 2
        ia[6] = 2
        ja[6] = 2
        ar[6] = 4.0  # a[2,2] = 4
        ia[7] = 3
        ja[7] = 2
        ar[7] = 2.0  # a[3,2] = 2
        ia[8] = 2
        ja[8] = 3
        ar[8] = 5.0  # a[2,3] = 5
        ia[9] = 3
        ja[9] = 3
        ar[9] = 6.0  # a[3,3] = 6

        glp.glp_load_matrix(lp, 9, ia, ja, ar)
        glp.glp_simplex(lp, smcp)

        Z = glp.glp_get_obj_val(lp)
        x1 = glp.glp_get_col_prim(lp, 1)
        x2 = glp.glp_get_col_prim(lp, 2)
        x3 = glp.glp_get_col_prim(lp, 3)

        self.assertAlmostEqual(Z, 733.3333, 4)
        self.assertAlmostEqual(x1, 33.3333, 4)
        self.assertAlmostEqual(x2, 66.6667, 4)
        self.assertAlmostEqual(x3, 0)

        glp.glp_delete_prob(lp)

    def test_nf_glpk(self):
        """Test the NetworkFlowGLPK interface to GLPK."""
        nf = NetworkFlowGLPK()

        nf.message_level = MessageLevel.ALL
        self.assertEqual(nf.message_level, MessageLevel.ALL)

        self.assertEqual(nf.simplex_method, SimplexMethod.PRIMAL)
        nf.simplex_method = SimplexMethod.DUAL
        self.assertEqual(nf.simplex_method, SimplexMethod.DUAL)

        limit = 1000
        nf.simplex_iteration_limit = limit
        self.assertEqual(nf.simplex_iteration_limit, limit)

        self.assertEqual(nf.status_code, glp.GLP_UNDEF)
        self.assertIn("UNDEF", nf.status_string)

        # TODO: How to use the NetworkFlowGLPK interface?
