old_contents: # coding: utf-8
  # Copyright (c) Pymatgen Development Team.
  # Distributed under the terms of the MIT License.
  
  from __future__ import division, unicode_literals
  
  """
  Created on Mar 18, 2012
  """
  
  
  __author__ = "Shyue Ping Ong"
  __copyright__ = "Copyright 2012, The Materials Project"
  __version__ = "0.1"
  __maintainer__ = "Shyue Ping Ong"
  __email__ = "shyue@mit.edu"
  __date__ = "Mar 18, 2012"
  
  import unittest2 as unittest
  import os
  
  from pymatgen.apps.borg.hive import VaspToComputedEntryDrone
  from pymatgen.apps.borg.queen import BorgQueen
  
  test_dir = os.path.join(os.path.dirname(__file__), "..", "..", "..", "..",
                          'test_files')
  
  
  class BorgQueenTest(unittest.TestCase):
  
      def setUp(self):
          drone = VaspToComputedEntryDrone()
          self.queen = BorgQueen(drone, test_dir, 1)
  
      def test_get_data(self):
          data = self.queen.get_data()
          self.assertEqual(len(data), 2)
  
      def test_load_data(self):
          drone = VaspToComputedEntryDrone()
          queen = BorgQueen(drone)
          queen.load_data(os.path.join(test_dir, "assimilated.json"))
          self.assertEqual(len(queen.get_data()), 1)
  
  if __name__ == "__main__":
      #import sys;sys.argv = ['', 'Test.testName']
      unittest.main()
  
new_contents: # coding: utf-8
  # Copyright (c) Pymatgen Development Team.
  # Distributed under the terms of the MIT License.
  
  from __future__ import division, unicode_literals
  
  """
  Created on Mar 18, 2012
  """
  
  
  __author__ = "Shyue Ping Ong"
  __copyright__ = "Copyright 2012, The Materials Project"
  __version__ = "0.1"
  __maintainer__ = "Shyue Ping Ong"
  __email__ = "shyue@mit.edu"
  __date__ = "Mar 18, 2012"
  
  import unittest2 as unittest
  import os
  
  from pymatgen.apps.borg.hive import VaspToComputedEntryDrone
  from pymatgen.apps.borg.queen import BorgQueen
  
  test_dir = os.path.join(os.path.dirname(__file__), "..", "..", "..", "..",
                          'test_files')
  
  
  class BorgQueenTest(unittest.TestCase):
  
      def setUp(self):
          drone = VaspToComputedEntryDrone()
          self.queen = BorgQueen(drone, test_dir, 1)
  
      def test_get_data(self):
          data = self.queen.get_data()
          self.assertEqual(len(data), 3)
  
      def test_load_data(self):
          drone = VaspToComputedEntryDrone()
          queen = BorgQueen(drone)
          queen.load_data(os.path.join(test_dir, "assimilated.json"))
          self.assertEqual(len(queen.get_data()), 1)
  
  if __name__ == "__main__":
      #import sys;sys.argv = ['', 'Test.testName']
      unittest.main()
  
subject: fix unit tesyt