Ejemplo n.º 1
0
 def test_default_memory_limit(self):
     utils.load_npy(self.npy_file)
     assert utils.get_memory_limit() == self.default_memory_limit
     memory_limit = utils.get_memory_limit()
     utils.set_memory_limit(1)
     utils.load_npy(self.npy_file)
     utils.set_memory_limit(memory_limit)
Ejemplo n.º 2
0
class TestLoadNpy():

    npy_file = 'tests/assets/array.npy'
    default_memory_limit = utils.get_memory_limit()

    def test_default_memory_limit(self):
        utils.load_npy(self.npy_file)
        assert utils.get_memory_limit() == self.default_memory_limit
        memory_limit = utils.get_memory_limit()
        utils.set_memory_limit(1)
        utils.load_npy(self.npy_file)
        utils.set_memory_limit(memory_limit)

    def test_custom_memory_limit(self):
        assert utils.get_memory_limit() == self.default_memory_limit
        utils.load_npy(self.npy_file, memory_limit=1)
        assert utils.get_memory_limit() == self.default_memory_limit
Ejemplo n.º 3
0
 def test_custom_memory_limit(self):
     assert utils.get_memory_limit() == self.default_memory_limit
     utils.load_npy(self.npy_file, memory_limit=1)
     assert utils.get_memory_limit() == self.default_memory_limit
Ejemplo n.º 4
0
 def test_set_memory_limit(self):
     default_memory_limit = utils.get_memory_limit()
     utils.set_memory_limit(1)
     assert utils.get_memory_limit() == 1
     utils.set_memory_limit(default_memory_limit)
Ejemplo n.º 5
0
 def test_default_memory_limit(self):
     assert utils.get_memory_limit() == 2 * 1024**3