def test_encode_without_permutation(self): corpus = 'hi my name is pat what is your name' d = DeLorean(corpus, bucket_size=2, n=1) self.assertEqual(d.encode_without_permutation('00'), ['hi', 'name']) self.assertEqual(d.encode_without_permutation('10'), ['my', 'name']) self.assertEqual(d.encode_without_permutation('11'), ['my', 'is'])
def test_encode_without_permutation_padding(self): corpus = 'hi my name is pat what is your name please tell me' d = DeLorean(corpus, bucket_size=4, n=2) self.assertEqual(d.encode_without_permutation('0'), ['hi']) self.assertEqual(d.encode_without_permutation('1'), ['name']) self.assertEqual(d.encode_without_permutation('000'), ['hi', 'pat']) self.assertEqual(d.encode_without_permutation('001'), ['hi', 'your'])
def test_decode_sorted(self): encoded = ['hi', 'is'] corpus = 'hi my name is pat what is your name' d = DeLorean(corpus, bucket_size=2, n=1) self.assertEqual(d.decode_sorted(['hi', 'name']), '00') self.assertEqual(d.decode_sorted(['my', 'is']), '11') self.assertEqual(d.decode_sorted(['hi', 'is']), '01')
def test_encode_nbs(self): corpus = 'hi my name is pat what is your name' d = DeLorean(corpus, bucket_size=2, n=1) self.assertEqual(d.encode_N_bs('0', 0), 'hi') self.assertEqual(d.encode_N_bs('1', 0), 'my') self.assertEqual(d.encode_N_bs('0', 1), 'name') self.assertEqual(d.encode_N_bs('1', 1), 'is')
def test_end_to_end(self): with open('script.txt', 'r') as f: corpus = f.read() d = DeLorean(corpus) for i in range(12): inp = ''.join(random.choice(['0', '1']) for _ in range(d.max_bits)) inp = inp[:-i] encoded = d.encode_without_permutation(inp) random.shuffle(encoded) decoded = d.decode(encoded) while len(inp) % d.n != 0: inp += '0' self.assertEqual(decoded, inp)
def incr_stat(name): if statsd_client: statsd_client.incr('puzzle.codes.%s' % name) app = Flask(__name__) sentry = Sentry(app) if 'SECRET_KEY' in os.environ: app.secret_key = os.environ['SECRET_KEY'] else: app.secret_key = 'test_secret_key' with open('script.txt', 'r') as f: app.delorean = DeLorean(f.read()) ANSWER_TEMPLATE = 'the %s of %s but like %s' COMPANIES_1 = [ 'airware', 'amazon', 'apple', 'atlassian', 'blend', 'box', 'cisco', 'de shaw', 'docker', 'dropbox', 'ebay', 'evernote', 'fidelity', 'github', 'groupon', 'hubspot', 'ibm', 'intuit', 'kayak', 'lastpass', 'logitech', 'mailchimp', 'microsoft', 'mozilla', 'nvidia', 'oracle', 'pandora', 'pinterest', 'quizlet', 'redhat', 'slack', 'spacex', 'stripe', 'thumbtack', 'tumblr', 'twitch', 'uber', 'valve', 'whatsapp', 'yahoo', 'zappos', 'zillow' ] INDUSTRIES = [ 'agriculture', 'art', 'beauty', 'clean tech', 'buildings', 'consulting', 'consumers', 'marketing', 'ecommerce', 'education', 'energy', 'fashion',
def test_wfc(self): corpus = 'hi my name is pat what is your name' words = ['hi', 'my', 'name', 'is', 'pat', 'what', 'is', 'your', 'name'] self.assertEqual(DeLorean.words_from_corpus(corpus), words)
def test_truncate_extra_words(self): words = ['hi', 'my', 'name', 'is', 'pat'] d = DeLorean('', bucket_size=2, n=1) result = d.truncate_extra_words(words) expected = ['hi', 'my', 'name', 'is'] self.assertEqual(result, expected)
def test_uwobfo(self): words = ['hi', 'my', 'name', 'is', 'pat', 'what', 'is', 'your', 'name'] result = DeLorean.unique_words_sorted_by_first_occurrence(words) expected = ['hi', 'my', 'name', 'is', 'pat', 'what', 'your'] self.assertEqual(result, expected)