Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
simplify test data and optimize imports
  • Loading branch information
somefreestring committed Jan 12, 2020
commit 86535558a46b436e782796500beb60ddb8265303
3 changes: 3 additions & 0 deletions test/config.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
import os

TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "data", "test.csv")
3 changes: 0 additions & 3 deletions test/unit_tests/summary/test__summary.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
import logging
import os
import unittest

import pandas as pd
Expand All @@ -9,8 +8,6 @@
logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data", "test.csv")


class SummaryTest(unittest.TestCase):

Expand Down
6 changes: 2 additions & 4 deletions test/z_component_tests/test__classification.py
Original file line number Diff line number Diff line change
@@ -1,21 +1,19 @@
import logging
import os
import unittest

import numpy as np
import pandas as pd
from sklearn.neural_network import MLPClassifier

import pandas_ml_utils as pdu
from pandas_ml_utils.model.features_and_labels.target_encoder import OneHotEncodedTargets
from pandas_ml_utils.constants import *
from pandas_ml_utils.model.features_and_labels.target_encoder import OneHotEncodedTargets
from test.config import TEST_FILE
from test.utils import SMA

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data", "test.csv")


class ClassificationTest(unittest.TestCase):
maxDiff = None
Expand Down
5 changes: 2 additions & 3 deletions test/z_component_tests/test__common.py
Original file line number Diff line number Diff line change
@@ -1,17 +1,16 @@
import logging
import os
import unittest

import pandas as pd

import pandas_ml_utils as pdu
from pandas_ml_utils.model.features_and_labels.features_and_labels_extractor import FeatureTargetLabelExtractor
from pandas_ml_utils.analysis.correlation_analysis import _sort_correlation
from pandas_ml_utils.model.features_and_labels.features_and_labels_extractor import FeatureTargetLabelExtractor
from pandas_ml_utils.model.fitting.train_test_data import make_training_data
from test.config import TEST_FILE

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)
TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data", "test.csv")


class ComponentTest(unittest.TestCase):
Expand Down
3 changes: 1 addition & 2 deletions test/z_component_tests/test__cross_fold.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,12 +10,11 @@
import pandas_ml_utils as pdu
from pandas_ml_utils.summary.binary_classification_summary import BinaryClassificationSummary
from pandas_ml_utils.utils.functions import fig_to_png_base64
from test.config import TEST_FILE

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data", "test.csv")


class CrossFoldTest(unittest.TestCase):

Expand Down
5 changes: 1 addition & 4 deletions test/z_component_tests/test__cross_validation.py
Original file line number Diff line number Diff line change
@@ -1,19 +1,16 @@
import logging
import os
import unittest

import numpy as np
import pandas as pd
from sklearn.model_selection import KFold
from sklearn.neural_network import MLPClassifier

import pandas_ml_utils as pdu
from test.config import TEST_FILE

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data", "test.csv")


class CrossValidationTest(unittest.TestCase):

Expand Down
7 changes: 1 addition & 6 deletions test/z_component_tests/test__fnl_extractor.py
Original file line number Diff line number Diff line change
@@ -1,21 +1,16 @@
import logging
import os
import unittest

import numpy as np
import pandas as pd
from sklearn.neural_network import MLPClassifier

import pandas_ml_utils as pdu
from pandas_ml_utils.constants import *
from pandas_ml_utils.summary.binary_classification_summary import BinaryClassificationSummary
from pandas_ml_utils.utils.functions import fig_to_png_base64
from test.config import TEST_FILE

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data", "test.csv")


class FeatureAndLabelsExtractorTest(unittest.TestCase):

Expand Down
4 changes: 1 addition & 3 deletions test/z_component_tests/test__hyper_parameter_tuning.py
Original file line number Diff line number Diff line change
@@ -1,18 +1,16 @@
import logging
import os
import unittest

import numpy as np
import pandas as pd
from sklearn.neural_network import MLPClassifier

import pandas_ml_utils as pdu
from test.config import TEST_FILE

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data", "test.csv")


class HyportParameterOptimizationTest(unittest.TestCase):

Expand Down
3 changes: 1 addition & 2 deletions test/z_component_tests/test__lazy_dataframe.py
Original file line number Diff line number Diff line change
@@ -1,15 +1,14 @@
import logging
import os
import unittest

import pandas as pd
from sklearn.neural_network import MLPClassifier

import pandas_ml_utils as pdu
from test.config import TEST_FILE

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)
TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data", "test.csv")


class LazyDataFrameTest(unittest.TestCase):
Expand Down
7 changes: 3 additions & 4 deletions test/z_component_tests/test__multi_model.py
Original file line number Diff line number Diff line change
@@ -1,20 +1,19 @@
import logging
import os
import unittest

import numpy as np
import pandas as pd
from sklearn.neural_network import MLPClassifier

import pandas_ml_utils as pdu
from pandas_ml_utils.constants import *
from pandas_ml_utils.model.features_and_labels.target_encoder import OneHotEncodedTargets
from test.config import TEST_FILE
from test.utils import SMA
from pandas_ml_utils.constants import *

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data", "test.csv")


class MultiModelTest(unittest.TestCase):
maxDiff = None
Expand Down
10 changes: 3 additions & 7 deletions test/z_component_tests/test__pre_processor.py
Original file line number Diff line number Diff line change
@@ -1,20 +1,16 @@
import logging
import os
import unittest

import numpy as np
import pandas as pd
from sklearn.neural_network import MLPClassifier
from pandas_ml_utils.constants import *

import pandas_ml_utils as pdu
from pandas_ml_utils.summary.binary_classification_summary import BinaryClassificationSummary
from pandas_ml_utils.utils.functions import fig_to_png_base64
from pandas_ml_utils.constants import *
from test.config import TEST_FILE

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data", "test.csv")


class PreprocessorTest(unittest.TestCase):

Expand Down
7 changes: 2 additions & 5 deletions test/z_component_tests/test__regression.py
Original file line number Diff line number Diff line change
@@ -1,18 +1,15 @@
import logging
import os
import unittest

import numpy as np
import pandas as pd
from sklearn.neural_network import MLPClassifier, MLPRegressor
from sklearn.neural_network import MLPRegressor

import pandas_ml_utils as pdu
from test.config import TEST_FILE

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data", "test.csv")


class RegressionTest(unittest.TestCase):

Expand Down
3 changes: 1 addition & 2 deletions test/z_component_tests/test__summary.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,11 @@
import pandas_ml_utils as pdu
from pandas_ml_utils.summary.binary_classification_summary import BinaryClassificationSummary
from pandas_ml_utils.utils.functions import fig_to_png_base64
from test.config import TEST_FILE

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data", "test.csv")


class ClassificationTest(unittest.TestCase):

Expand Down