Merge branch 'dev' of https://github.com/datamllab/rlcard into dev

.gitignore

@@ -14,3 +14,4 @@ sonar-project.properties
 docs/rst
 docs/sphinx
 experiments/
+newtest/

.travis.yml

@@ -6,6 +6,6 @@ before_script:
   - pip install python-coveralls
   - pip install pytest-cover
 script: 
-  - py.test test/test_utils.py --cov=rlcard/utils/
+  - py.test test/ --cov=rlcard
 after_success:
   - coveralls

rlcard/agents/dqn_agent.py

@@ -32,7 +32,7 @@
 import rlcard
 from rlcard.utils.utils import *
 
-Transition = namedtuple('Transition', '"state', 'action', 'reward', 'next_state', 'done'])
+Transition = namedtuple("Transition", ["state", "action", "reward", "next_state", "done"])
 
 class DQNAgent(object):
 

rlcard/utils/logger.py

@@ -91,3 +91,11 @@ def make_plot(self, save_path = ''):
             os.makedirs(save_dir)
 
         fig.savefig(save_path)
+
+    def close_file(self):
+        ''' Close the created file objects
+        '''
+        if self.log_path != None:
+            self.log_file.close()
+        if self.csv_path != None:
+            self.csv_file.close()

test/utils/test_logger.py

@@ -0,0 +1,34 @@
+import unittest
+import random
+import numpy as np
+import os
+from rlcard.utils.logger import Logger
+
+class TestLoggerMethos(unittest.TestCase):
+
+    def test_log(self):
+        logger = Logger(xlabel="x", ylabel="y", legend="test", log_path="./newtest/test_log.txt")
+        logger.log("test text")
+        f = open("./newtest/test_log.txt", "r")
+        contents = f.read()
+        self.assertEqual(contents, "test text\n")
+        logger.close_file()
+
+    def test_add_point(self):
+        logger = Logger(xlabel="x", ylabel="y", legend="test", csv_path="./newtest/test_csv.csv")
+        logger.add_point(x=1, y=1)
+        self.assertEqual(logger.xs[0], 1)
+        self.assertEqual(logger.ys[0], 1)
+
+    def test_make_plot(self):
+        logger = Logger(xlabel="x", ylabel="y", legend="test")
+        for x in range(10):
+            logger.add_point(x=x, y=x*x)
+        logger.make_plot(save_path='./newtest/test.png')
+
+    def test_close_file(self):
+        logger = Logger(xlabel="x", ylabel="y", legend="test", log_path="./newtest/test_log.txt",csv_path="./newtest/test_csv.csv")
+        logger.close_file()
+
+if __name__ == '__main__':
+    unittest.main()