Python get_memory_limitの例

コード例 #1

 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)

コード例 #2

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

コード例 #3

 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

コード例 #4

 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)

コード例 #5

 def test_default_memory_limit(self):
     assert utils.get_memory_limit() == 2 * 1024**3