# Path: scgv/models/model.py
# class DataModel(BaseModel):
# 
#     def __init__(self, filename):
#         self.data = DataLoader(filename)
#         self.data.load()
#         self.data.filter_cells()
#         self.data.order_cells()
# 
#         super(DataModel, self).__init__(self.data)
# 
# Path: scgv/models/featuremat_model.py
# class FeaturematModel(ModelDelegate):
# 
#     def __init__(self, model):
#         super(FeaturematModel, self).__init__(model)
#         assert self.model.data.features_df is not None
#         assert self.model.data.featuremat_df is not None
#         self.heatmap = self.model.make_featuremat(ordering=self.ordering) + 2.0


import numpy as np
from scgv.models.model import DataModel  # , gate_compare
from scgv.models.featuremat_model import FeaturematModel
'''
Created on Dec 21, 2016

@author: lubo
'''
# import pytest


def test_data_model_make_01(example_data):
    model = DataModel(example_data)
    assert model is not None
    model.make()

    sectors_legend = model.make_sectors_legend()
    print(sectors_legend)


def test_make_featuremat(example_data):

# Path: scgv/models/model.py
# class DataModel(BaseModel):
# 
#     def __init__(self, filename):
#         self.data = DataLoader(filename)
#         self.data.load()
#         self.data.filter_cells()
#         self.data.order_cells()
# 
#         super(DataModel, self).__init__(self.data)
# 
# Path: scgv/models/featuremat_model.py
# class FeaturematModel(ModelDelegate):
# 
#     def __init__(self, model):
#         super(FeaturematModel, self).__init__(model)
#         assert self.model.data.features_df is not None
#         assert self.model.data.featuremat_df is not None
#         self.heatmap = self.model.make_featuremat(ordering=self.ordering) + 2.0


# Path: scgv/tests/test_data_model.py
import numpy as np
from scgv.models.model import DataModel  # , gate_compare
from scgv.models.featuremat_model import FeaturematModel

'''
Created on Dec 21, 2016

@author: lubo
'''
# import pytest


def test_data_model_make_01(example_data):
    model = DataModel(example_data)
    assert model is not None
    model.make()

    sectors_legend = model.make_sectors_legend()
    print(sectors_legend)


def test_make_featuremat(example_data):

    model = DataModel(example_data)