import unittest

import pytest

import numpy as np

import pandas as pd

from mofapy2.run.entry_point import entry_point

class TestBuildModel:

    def test_build_basic(self):

        ent = entry_point()

        ent.set_data_options(scale_groups=False, scale_views=False)

        views_names = ["view1", "view2"]

        groups_names = ["groupA", "groupB"]

        # Set dimensions

        n_g1, n_g2 = 10, 20

        d_m1, d_m2 = 30, 40

        np.random.seed(42)

        ent.set_data_matrix(

            [

                [np.random.random((n_g1, d_m1)), np.random.random((n_g2, d_m1))],

                [np.random.random((n_g1, d_m2)), np.random.random((n_g2, d_m2))],

            ]

        )

        ent.set_model_options()

        ent.set_train_options()

        ent.build()

if __name__ == "__main__":

    unittest.main()