def test_filename(self):
     """Verify that an appropriate file name is generated in an appropriate
     folder"""
     file_path = tempfile.mkdtemp() + os.path.join('some', 'depth', 'here')
     cp = Checkpointer(file_path)
     cp.counter = 25
     filename = cp._filename('A WeIrD TaG')
     self.assertTrue(os.path.join('some', 'depth', 'here') in filename)
     self.assertTrue('25' in filename)
     self.assertTrue('aweirdtag' in filename)
Example #2
0
 def test_filename(self):
     """Verify that an appropriate file name is generated in an appropriate
     folder"""
     file_path = tempfile.mkdtemp() + os.path.join('some', 'depth', 'here')
     cp = Checkpointer(file_path)
     cp.counter = 25
     filename = cp._filename('A WeIrD TaG')
     self.assertTrue(os.path.join('some', 'depth', 'here') in filename)
     self.assertTrue('25' in filename)
     self.assertTrue('aweirdtag' in filename)
    def test_basic_serialization(self):
        """We should be able to store and retrieve an object. Verify that this
        is occurring outside of local memory by comparing to the original."""
        to_store = {"some": "value", 123: 456}
        cp = Checkpointer(tempfile.mkdtemp())
        cp.counter = 1
        cp._serialize("a-tag", to_store)
        to_store["some"] = "other"
        result = cp._deserialize("a-tag")
        self.assertEqual(result, {"some": "value", 123: 456})
        self.assertEqual(to_store, {"some": "other", 123: 456})

        cp.counter = 2
        cp._serialize("a-tag", to_store)
        to_store["some"] = "more"
        result = cp._deserialize("a-tag")
        self.assertEqual(result, {"some": "other", 123: 456})
        self.assertEqual(to_store, {"some": "more", 123: 456})
        cp.counter = 1
        result = cp._deserialize("a-tag")
        self.assertEqual(result, {"some": "value", 123: 456})
Example #4
0
    def test_basic_serialization(self):
        """We should be able to store and retrieve an object. Verify that this
        is occurring outside of local memory by comparing to the original."""
        to_store = {"some": "value", 123: 456}
        cp = Checkpointer(tempfile.mkdtemp())
        cp.counter = 1
        cp._serialize("a-tag", to_store)
        to_store["some"] = "other"
        result = cp._deserialize("a-tag")
        self.assertEqual(result, {"some": "value", 123: 456})
        self.assertEqual(to_store, {"some": "other", 123: 456})

        cp.counter = 2
        cp._serialize("a-tag", to_store)
        to_store["some"] = "more"
        result = cp._deserialize("a-tag")
        self.assertEqual(result, {"some": "other", 123: 456})
        self.assertEqual(to_store, {"some": "more", 123: 456})
        cp.counter = 1
        result = cp._deserialize("a-tag")
        self.assertEqual(result, {"some": "value", 123: 456})