いろいろ変更

[stux/ultron.git] / src / test / Python / TestDataReader.py

diff --git a/src/test/Python/TestDataReader.py b/src/test/Python/TestDataReader.py
index 434a801..68a12f6 100644 (file)
--- a/src/test/Python/TestDataReader.py
+++ b/src/test/Python/TestDataReader.py
@@ -11,17 +11,13 @@ class TestDataReader(unittest.TestCase):
         print("データ読み込みパス："+os.path.dirname(os.path.abspath(__file__)))
         print("データファイル:" + os.path.abspath(PATH))
         data_reader = DataReader.DataReader()
-        res = data_reader.read(os.path.abspath("%s" % PATH))
+        res = data_reader._read(os.path.abspath("%s" % PATH))
         assert len(res.getData()) == 10, "データサイズが不正です"
         assert res.getLable() == 3, res.getLable() + "は期待通りの値ではありません。"
 
-    def test_walk_data_directory(self):
-        data_reader = DataReader.DataReader()
-        data_reader.walkDataDirectory(os.path.abspath("..\\resources\\data\\2007"))
-
     def test_create_learning_data(self):
-        data_reader = DataReader.DataReader()
-        result=data_reader.createLearningData()
+        data_reader: DataReader = DataReader.DataReader()
+        result=data_reader.parse_data_files()
         print(result[0].getData())
 
         assert len(result) == 3, str(len(result))+"データ個数が不正です"