FilesExpand file tree

eval_example

/

test_supervised_task.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

35 lines (25 loc) · 868 Bytes

eval_example

/

test_supervised_task.py

Copy path

Top

File metadata and controls

• Code
• Blame

35 lines (25 loc) · 868 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

from typing import Tuple

import unittest

import numpy as np

from openml.tasks import get_task

from .test_task import OpenMLTaskTest

class OpenMLSupervisedTaskTest(OpenMLTaskTest):

"""

A helper class. The methods of the test case

are only executed in subclasses of the test case.

"""

__test__ = False

@classmethod

def setUpClass(cls):

if cls is OpenMLSupervisedTaskTest:

raise unittest.SkipTest(

"Skip OpenMLSupervisedTaskTest tests,"

" it's a base class"

)

super(OpenMLSupervisedTaskTest, cls).setUpClass()

def setUp(self, n_levels: int = 1):

super(OpenMLSupervisedTaskTest, self).setUp()

def test_get_X_and_Y(self) -> Tuple[np.ndarray, np.ndarray]:

task = get_task(self.task_id)

X, Y = task.get_X_and_y()

return X, Y