def test_add_memory_gpu(self):
"""Test that we can correctly add memory to a GPU moemory array"""
# We create an initial GPU
gpu = GPU("NVIDIA", "GF110GL")
self.assertEquals(0, len(gpu.memory))
# We add a Memory element
memory = Memory(1111, "kilobytes")
gpu.add_memory(memory)
self.assertEquals(1, len(gpu.memory))
self.assertEquals(memory, gpu.memory[0])
# We verify that we can not add memory of the wrong type
gpu.add_memory("xxx")
self.assertEquals(1, len(gpu.memory))
self.assertEquals(memory, gpu.memory[0])
def test_remove_memory(self):
"""We verify it is possible to delete a memory element"""
# We create an initial GPU
gpu = GPU("NVIDIA", "GF110GL")
print(len(gpu.memory))
self.assertEquals(0, len(gpu.memory))
# We add a Memory element
memory = Memory(1111, "kilobytes")
gpu.add_memory(memory)
self.assertEquals(1, len(gpu.memory))
# We remove it
gpu.remove_memory(memory)
self.assertEquals(0, len(gpu.memory))
# We verify that removing an inexistent elements gives no error
gpu.remove_memory("xxx")
self.assertEquals(0, len(gpu.memory))