def test_usage(self):
        beat = Heartbeat()

        public_beat = beat.get_public()

        with open('files/test.txt', 'rb') as file:
            (tag, state) = beat.encode(file)

        challenge = beat.gen_challenge(state)

        with open('files/test.txt', 'rb') as file:
            proof = public_beat.prove(file, challenge, tag)

        is_valid = beat.verify(proof, challenge, state)
        if (is_valid):
            print('file is stored by the server')
        else:
            print('file proof invalid')

        self.assertTrue(is_valid)
    def test_usage(self):
        beat = Heartbeat()

        public_beat = beat.get_public()

        with open("files/test.txt", "rb") as file:
            (tag, state) = beat.encode(file)

        challenge = beat.gen_challenge(state)

        with open("files/test.txt", "rb") as file:
            proof = public_beat.prove(file, challenge, tag)

        is_valid = beat.verify(proof, challenge, state)
        if is_valid:
            print("file is stored by the server")
        else:
            print("file proof invalid")

        self.assertTrue(is_valid)